1 files changed, 8 insertions, 7 deletions
diff --git a/src/test/scala/chiselTests/NamingAnnotationTest.scala b/src/test/scala/chiselTests/NamingAnnotationTest.scala
index a02dfe56..41cba1de 100644
--- a/src/test/scala/chiselTests/NamingAnnotationTest.scala
+++ b/src/test/scala/chiselTests/NamingAnnotationTest.scala
@@ -5,6 +5,7 @@ package chiselTests
 import chisel3._
 import chisel3.experimental.chiselName
 import chisel3.internal.InstanceId
+import chisel3.stage.ChiselStage
 
 import scala.collection.mutable.ListBuffer
 
@@ -102,7 +103,7 @@ class NamedModule extends NamedModuleTester {
   val test = expectName(FunctionMockup(), "test")
   val test2 = expectName(test +& 2.U, "test2")
   val test3 = expectName(ImplicitlyNamed(), "test3")
-  
+
   val test4 = new NonModule
   expectName(test4.value, "test4_value")
   expectName(test4.inner.value, "test4_inner_value")
@@ -237,30 +238,30 @@ class NamingAnnotationSpec extends ChiselPropSpec {
   property("NamedModule should have function hierarchical names") {
     // TODO: clean up test style
     var module: NamedModule = null
-    elaborate { module = new NamedModule; module }
+    ChiselStage.elaborate { module = new NamedModule; module }
     assert(module.getNameFailures() == Nil)
   }
 
   property("NameCollisionModule should disambiguate collisions") {
     // TODO: clean up test style
     var module: NameCollisionModule = null
-    elaborate { module = new NameCollisionModule; module }
+    ChiselStage.elaborate { module = new NameCollisionModule; module }
     assert(module.getNameFailures() == Nil)
   }
 
   property("PartialNamedModule should have partial names") {
     // TODO: clean up test style
     var module: PartialNamedModule = null
-    elaborate { module = new PartialNamedModule; module }
+    ChiselStage.elaborate { module = new PartialNamedModule; module }
     assert(module.getNameFailures() == Nil)
   }
 
   property("NonNamedModule should elaborate") {
-    elaborate { new NonNamedModule }
+    ChiselStage.elaborate { new NonNamedModule }
   }
 
   property("NonNamedFunction should elaborate") {
-    elaborate { new NonNamedFunction }
+    ChiselStage.elaborate { new NonNamedFunction }
   }
 
   property("NonBuilderFunction should run outside a Builder context") {
@@ -269,7 +270,7 @@ class NamingAnnotationSpec extends ChiselPropSpec {
 
   property("NoChiselNamePrefix should prevent prefixing when using @chiselName") {
     var module: NoChiselNamePrefixTester = null
-    elaborate { module = new NoChiselNamePrefixTester; module }
+    ChiselStage.elaborate { module = new NoChiselNamePrefixTester; module }
     assert(module.getNameFailures().isEmpty)
   }
 }