1 files changed, 6 insertions, 5 deletions
diff --git a/src/test/scala/chiselTests/MissingCloneBindingExceptionSpec.scala b/src/test/scala/chiselTests/MissingCloneBindingExceptionSpec.scala
index 50dbb341..856a91f7 100644
--- a/src/test/scala/chiselTests/MissingCloneBindingExceptionSpec.scala
+++ b/src/test/scala/chiselTests/MissingCloneBindingExceptionSpec.scala
@@ -2,12 +2,13 @@
 
 package chiselTests
 import Chisel.ChiselException
+import chisel3.stage.ChiselStage
 import org.scalatest._
 import org.scalatest.matchers.should.Matchers
 
-class MissingCloneBindingExceptionSpec extends ChiselFlatSpec with Matchers {
+class MissingCloneBindingExceptionSpec extends ChiselFlatSpec with Matchers with Utils {
   behavior of "missing cloneType in Chisel3"
-  ( the[ChiselException] thrownBy {
+  ( the [ChiselException] thrownBy extractCause[ChiselException] {
     import chisel3._
 
     class Test extends Module {
@@ -26,11 +27,11 @@ class MissingCloneBindingExceptionSpec extends ChiselFlatSpec with Matchers {
       })
     }
 
-    elaborate(new TestTop)
+    ChiselStage.elaborate(new TestTop)
   }).getMessage should include("make all parameters immutable")
 
   behavior of "missing cloneType in Chisel2"
-  ( the[ChiselException] thrownBy {
+  ( the [ChiselException] thrownBy extractCause[ChiselException] {
     import Chisel._
 
     class Test extends Module {
@@ -49,6 +50,6 @@ class MissingCloneBindingExceptionSpec extends ChiselFlatSpec with Matchers {
       }
     }
 
-    elaborate(new TestTop)
+    ChiselStage.elaborate(new TestTop)
   }).getMessage should include("make all parameters immutable")
 }