2 files changed, 6 insertions, 5 deletions

diff --git a/bittorrent.cabal b/bittorrent.cabal
index 292680dd..0ceec550 100644
--- a/bittorrent.cabal
+++ b/bittorrent.cabal
@@ -50,7 +50,6 @@ library
   hs-source-dirs:      src
   exposed-modules:     Data.Torrent
                        Data.Torrent.Progress
-                       Data.Torrent.Tree
                        Network.BitTorrent
                        Network.BitTorrent.Address
                        Network.BitTorrent.Client
@@ -87,11 +86,13 @@ library
                        Network.BitTorrent.Internal.Cache
                        Network.BitTorrent.Internal.Types
                        System.Torrent.FileMap
+                       System.Torrent.Tree
   else
     other-modules:
                        Network.BitTorrent.Internal.Cache
                        Network.BitTorrent.Internal.Types
                        System.Torrent.FileMap
+                       System.Torrent.Tree
 
   build-depends:       base       == 4.*
                      , lifted-base
diff --git a/src/Data/Torrent/Tree.hs b/src/System/Torrent/Tree.hs
index 5825422f..41cfb360 100644
--- a/src/Data/Torrent/Tree.hs
+++ b/src/System/Torrent/Tree.hs
@@ -10,7 +10,7 @@
 {-# LANGUAGE FlexibleInstances  #-}
 {-# LANGUAGE TemplateHaskell    #-}
 {-# LANGUAGE DeriveDataTypeable #-}
-module Data.Torrent.Tree
+module System.Torrent.Tree
        ( -- * Directory tree
          DirTree (..)
 
@@ -18,7 +18,7 @@ module Data.Torrent.Tree
        , build
 
          -- * Query
-       , Data.Torrent.Tree.lookup
+       , System.Torrent.Tree.lookup
        , lookupDir
        , fileNumber
        , dirNumber
@@ -61,13 +61,13 @@ build MultiFile {..} = Dir $ M.singleton liDirName files
 lookup :: [FilePath] -> DirTree a -> Maybe (DirTree a)
 lookup []        t      = Just t
 lookup (p : ps) (Dir m) | Just subTree <- M.lookup (BC.pack p) m
-                        = Data.Torrent.Tree.lookup ps subTree
+                        = System.Torrent.Tree.lookup ps subTree
 lookup _         _      = Nothing
 
 -- | Lookup directory by path.
 lookupDir :: [FilePath] -> DirTree a -> Maybe [(ByteString, DirTree a)]
 lookupDir ps d = do
-  subTree <- Data.Torrent.Tree.lookup ps d
+  subTree <- System.Torrent.Tree.lookup ps d
   case subTree of
     File _  -> Nothing
     Dir  es -> Just $ M.toList es