diff --git a/src/test/scala/lazabs/horn/Util.scala b/src/test/scala/lazabs/horn/Util.scala
index d9060254..ff9fbfd5 100644
--- a/src/test/scala/lazabs/horn/Util.scala
+++ b/src/test/scala/lazabs/horn/Util.scala
@@ -1,6 +1,6 @@
 package lazabs.horn
 
-import lazabs.horn.bottomup.HornWrapper
+import lazabs.horn.Util._
 import ap.api.SimpleAPI.ProverStatus
 import lazabs.horn.preprocessor.HornPreprocessor.{CounterExample, Solution}
 import org.scalatest.matchers.{MatchResult, Matcher}
@@ -24,8 +24,8 @@ trait CHCResultMatchers extends Matchers {
   }
 
   def hideOutput[A](comp : => A) : A =
-    Console.withOut(HornWrapper.NullStream) {
-      Console.withErr(HornWrapper.NullStream) {
+    Console.withOut(NullStream) {
+      Console.withErr(NullStream) {
         comp
       }
     }
diff --git a/src/test/scala/lazabs/horn/theories/Rationals.scala b/src/test/scala/lazabs/horn/theories/Rationals.scala
index 2402ee03..cfb799a4 100644
--- a/src/test/scala/lazabs/horn/theories/Rationals.scala
+++ b/src/test/scala/lazabs/horn/theories/Rationals.scala
@@ -29,6 +29,7 @@
 
 package lazabs.horn.theories
 
+import lazabs.horn.HornWrapper
 import lazabs.horn.bottomup._
 import ap.parser._
 import ap.theories.rationals.Rationals