diff --git a/registry-hedgehog.cabal b/registry-hedgehog.cabal
--- a/registry-hedgehog.cabal
+++ b/registry-hedgehog.cabal
@@ -4,10 +4,10 @@
 --
 -- see: https://github.com/sol/hpack
 --
--- hash: 8da4d7249d8800f06d303acbc3ed33c9d559db7bfde2ba78d8a2cf4293c4d2d0
+-- hash: 7d0b0684f2ba9eb6cb838ca63d1f08de7ee191046c523b7d0ede13b888d63a06
 
 name:           registry-hedgehog
-version:        0.7.0.4
+version:        0.7.0.5
 synopsis:       utilities to work with Hedgehog generators and `registry`
 description:    This library provides some functions to extract generators from a "Registry" and make stateful modifications of that Registry to precisely control the generation of data
 category:       Control
diff --git a/test/Test/Data/Registry/HedgehogSpec.hs b/test/Test/Data/Registry/HedgehogSpec.hs
--- a/test/Test/Data/Registry/HedgehogSpec.hs
+++ b/test/Test/Data/Registry/HedgehogSpec.hs
@@ -9,6 +9,7 @@
 import Control.Monad.Morph (hoist)
 import Data.Registry
 import Data.Registry.Hedgehog
+import Data.Registry.State
 import qualified Data.Text as T
 import qualified Data.Text.IO as T
 import Hedgehog.Gen as Gen
diff --git a/test/Test/Tutorial/Exercise5.hs b/test/Test/Tutorial/Exercise5.hs
--- a/test/Test/Tutorial/Exercise5.hs
+++ b/test/Test/Tutorial/Exercise5.hs
@@ -7,6 +7,7 @@
 
 import Data.Registry
 import Data.Registry.Hedgehog
+import Data.Registry.State
 import Data.Text as T
 import Hedgehog hiding (test)
 import Protolude