1 files changed, 3 insertions, 5 deletions
diff --git a/src/test/scala/firrtlTests/annotationTests/TargetDirAnnotationSpec.scala b/src/test/scala/firrtlTests/annotationTests/TargetDirAnnotationSpec.scala
index 60cbf0fc..eb061d8f 100644
--- a/src/test/scala/firrtlTests/annotationTests/TargetDirAnnotationSpec.scala
+++ b/src/test/scala/firrtlTests/annotationTests/TargetDirAnnotationSpec.scala
@@ -14,11 +14,9 @@ class FindTargetDirTransform(expected: String) extends Transform {
   var run = false
   def execute(state: CircuitState): CircuitState = {
     run = true
-    state.annotations.foreach { aMap =>
-      aMap.annotations.collectFirst {
-        case TargetDirAnnotation(expected) =>
-          foundTargetDir = true
-      }
+    state.annotations.collectFirst {
+      case TargetDirAnnotation(expected) =>
+        foundTargetDir = true
     }
     state
   }