1 files changed, 6 insertions, 6 deletions
diff --git a/src/test/scala/chiselTests/OptionBundle.scala b/src/test/scala/chiselTests/OptionBundle.scala
index 1b7be1ec..b23c1f6b 100644
--- a/src/test/scala/chiselTests/OptionBundle.scala
+++ b/src/test/scala/chiselTests/OptionBundle.scala
@@ -27,21 +27,21 @@ class OptionBundleModule(hasIn: Boolean) extends Module {
 class SomeOptionBundleTester(expected: Boolean) extends BasicTester {
   val mod = Module(new OptionBundleModule(true))
   mod.io.in.get := Bool(expected)
-  io.error := mod.io.out != Bool(expected)
-  io.done := Bool(true)
+  assert(mod.io.out === Bool(expected))
+  stop()
 }
 
 class NoneOptionBundleTester() extends BasicTester {
   val mod = Module(new OptionBundleModule(true))
-  io.error := mod.io.out != Bool(false)
-  io.done := Bool(true)
+  assert(mod.io.out === Bool(false))
+  stop()
 }
 
 class InvalidOptionBundleTester() extends BasicTester {
   val mod = Module(new OptionBundleModule(false))
   mod.io.in.get := Bool(true)
-  io.error := UInt(1)
-  io.done := Bool(true)
+  assert(Bool(false))
+  stop()
 }
 
 class OptionBundleSpec extends ChiselFlatSpec {