2 files changed, 2 insertions, 2 deletions

diff --git a/tests/System/Torrent/FileMapSpec.hs b/tests/System/Torrent/FileMapSpec.hs
index 36632b3e..85180c0a 100644
--- a/tests/System/Torrent/FileMapSpec.hs
+++ b/tests/System/Torrent/FileMapSpec.hs
@@ -9,7 +9,7 @@ import System.FilePath
 import System.IO.Unsafe
 import Test.Hspec
 
-import Data.Torrent.Layout
+import Data.Torrent
 import System.Torrent.FileMap as FM
 
 
diff --git a/tests/System/Torrent/StorageSpec.hs b/tests/System/Torrent/StorageSpec.hs
index 30322545..ebf4fe3e 100644
--- a/tests/System/Torrent/StorageSpec.hs
+++ b/tests/System/Torrent/StorageSpec.hs
@@ -8,7 +8,7 @@ import System.IO.Unsafe
 import Test.Hspec
 
 import Data.Torrent.Bitfield as BF
-import Data.Torrent.Layout
+import Data.Torrent
 import Data.Torrent.Piece
 import System.Torrent.Storage