1 files changed, 4 insertions, 3 deletions
diff --git a/src/test/scala/firrtlTests/annotationTests/JsonProtocolSpec.scala b/src/test/scala/firrtlTests/annotationTests/JsonProtocolSpec.scala
index 2dce89db..507c5291 100644
--- a/src/test/scala/firrtlTests/annotationTests/JsonProtocolSpec.scala
+++ b/src/test/scala/firrtlTests/annotationTests/JsonProtocolSpec.scala
@@ -14,7 +14,8 @@ case class AnAnnotation(
     port: Port,
     statement: Statement,
     expr: Expression,
-    tpe: Type
+    tpe: Type,
+    groundType: GroundType
 ) extends NoTargetAnnotation
 
 class JsonProtocolSpec extends FlatSpec with Matchers {
@@ -33,10 +34,10 @@ class JsonProtocolSpec extends FlatSpec with Matchers {
     val stmt = mod.asInstanceOf[Module].body
     val expr = stmt.asInstanceOf[Block].stmts.head.asInstanceOf[Connect].expr
     val tpe = port.tpe
-    val inputAnnos = Seq(AnAnnotation(cir.info, cir, mod, port, stmt, expr, tpe))
+    val groundType = port.tpe.asInstanceOf[GroundType]
+    val inputAnnos = Seq(AnAnnotation(cir.info, cir, mod, port, stmt, expr, tpe, groundType))
     val annosString = JsonProtocol.serialize(inputAnnos)
     val outputAnnos = JsonProtocol.deserialize(annosString)
     inputAnnos should be (outputAnnos)
   }
-
 }