Some cleanup

1 files changed, 3 insertions, 3 deletions

diff --git a/src/main/scala/firrtl/annotations/JsonProtocol.scala b/src/main/scala/firrtl/annotations/JsonProtocol.scala
index 92cf4264..4fb40b76 100644
--- a/src/main/scala/firrtl/annotations/JsonProtocol.scala
+++ b/src/main/scala/firrtl/annotations/JsonProtocol.scala
@@ -10,7 +10,7 @@ import org.json4s.native.Serialization
 import org.json4s.native.Serialization.{read, writePretty}
 
 trait HasSerializationHints {
-  // For serialization of complicated constuctor arguments, let the annotation
+  // For serialization of complicated constructor arguments, let the annotation
   // writer specify additional type hints for relevant classes that might be
   // contained within
   def typeHints: Seq[Class[_]]
@@ -93,10 +93,10 @@ object JsonProtocol {
   def serialize(annos: Seq[Annotation]): String = serializeTry(annos).get
 
   def serializeTry(annos: Seq[Annotation]): Try[String] = {
-    val tags = annos.collect({
+    val tags = annos.flatMap({
       case anno: HasSerializationHints => anno.getClass +: anno.typeHints
       case anno => Seq(anno.getClass)
-    }).flatten.distinct
+    }).distinct
 
     implicit val formats = jsonFormat(tags)
     Try(writePretty(annos))