testing improvements

1 files changed, 2 insertions, 2 deletions

diff --git a/src/test/scala/chiselTests/ChiselSpec.scala b/src/test/scala/chiselTests/ChiselSpec.scala
index fd602ef3..002ace8c 100644
--- a/src/test/scala/chiselTests/ChiselSpec.scala
+++ b/src/test/scala/chiselTests/ChiselSpec.scala
@@ -35,11 +35,11 @@ import org.scalatest.prop._
 import org.scalacheck._
 import Chisel.testers._
 
-class ChiselSpec extends FlatSpec with PropertyChecks {
+class ChiselPropSpec extends PropSpec with PropertyChecks {
   def execute(t: => BasicTester): Boolean = TesterDriver.execute(t)
 
   val safeUIntWidth = Gen.choose(1, 31) 
   val safeUInts = Gen.choose(0, (1 << 30))
-
+  val vecSizes = Gen.choose(0, 4)
 }