diff --git a/bench/Bench.hs b/bench/Bench.hs
index 93bf144..1880492 100644
--- a/bench/Bench.hs
+++ b/bench/Bench.hs
@@ -1,6 +1,7 @@
 import Criterion.Main
 import Language.Haskell.GhcMod.Target
-import TestUtils
+import Language.Haskell.GhcMod.Monad
+import Language.Haskell.GhcMod.Types
 import Dir
 import System.IO.Temp
 import System.Process hiding (env)