diff --git a/src/main/scala/barstools/tapeout/transforms/GenerateModelStageMain.scala b/src/main/scala/barstools/tapeout/transforms/GenerateModelStageMain.scala
index 2de98408..c48d4c35 100644
--- a/src/main/scala/barstools/tapeout/transforms/GenerateModelStageMain.scala
+++ b/src/main/scala/barstools/tapeout/transforms/GenerateModelStageMain.scala
@@ -11,7 +11,7 @@ import firrtl.transforms.BlackBoxResourceFileNameAnno
 import logger.LazyLogging
 
 private class GenerateModelStageMain(annotations: AnnotationSeq) extends LazyLogging {
-  val outAnno:     Option[String] = annotations.collectFirst { case OutAnnoAnnotation(s) => s }
+  val outAnno: Option[String] = annotations.collectFirst { case OutAnnoAnnotation(s) => s }
 
   val annoFiles: List[String] = annotations.flatMap {
     case InputAnnotationFileAnnotation(f) => Some(f)
diff --git a/src/main/scala/barstools/tapeout/transforms/stage/TapeoutStage.scala b/src/main/scala/barstools/tapeout/transforms/stage/TapeoutStage.scala
index bff297b1..7ab59717 100644
--- a/src/main/scala/barstools/tapeout/transforms/stage/TapeoutStage.scala
+++ b/src/main/scala/barstools/tapeout/transforms/stage/TapeoutStage.scala
@@ -32,7 +32,7 @@ trait TapeoutCli {
   parser.note("Tapeout specific options")
 
   Seq(
-    OutAnnoAnnotation,
+    OutAnnoAnnotation
   ).foreach(_.addOptions(parser))
 }
 
diff --git a/src/test/scala/barstools/tapeout/transforms/GenerateSpec.scala b/src/test/scala/barstools/tapeout/transforms/GenerateSpec.scala
index a4dcd780..bb089de9 100644
--- a/src/test/scala/barstools/tapeout/transforms/GenerateSpec.scala
+++ b/src/test/scala/barstools/tapeout/transforms/GenerateSpec.scala
@@ -91,7 +91,7 @@ class GenerateSpec extends AnyFreeSpec {
     val targetDir = "test_run_dir/generate_spec"
     generateTestData(targetDir)
 
-    GenerateTop.main(
+    GenerateModelStageMain.main(
       Array(
         "-i",
         s"$targetDir/GenerateExampleTester.fir",