diff --git a/clash-protocols/src/Protocols/Hedgehog.hs b/clash-protocols/src/Protocols/Hedgehog.hs
index af2296f..b6188ba 100644
--- a/clash-protocols/src/Protocols/Hedgehog.hs
+++ b/clash-protocols/src/Protocols/Hedgehog.hs
@@ -55,9 +55,6 @@ import qualified Hedgehog.Range as Range
 -- lifted-async
 import Control.Concurrent.Async.Lifted (race)
 
--- me
-import Protocols.Hedgehog.Types
-
 -- | Whether to stall or not. Used in 'idWithModel'.
 data StallMode = NoStall | Stall
   deriving (Show, Enum, Bounded)
diff --git a/clash-protocols/src/Protocols/Hedgehog/Internal.hs b/clash-protocols/src/Protocols/Hedgehog/Internal.hs
index 19846d5..2a2df4f 100644
--- a/clash-protocols/src/Protocols/Hedgehog/Internal.hs
+++ b/clash-protocols/src/Protocols/Hedgehog/Internal.hs
@@ -9,7 +9,10 @@
 {- |
 Internals for "Protocols.Hedgehog".
 -}
-module Protocols.Hedgehog.Internal where
+module Protocols.Hedgehog.Internal (
+  module Protocols.Hedgehog.Internal,
+  module Protocols.Hedgehog.Types,
+) where
 
 -- base
 import Data.Proxy (Proxy (Proxy))