2 files changed, 4 insertions, 2 deletions

diff --git a/src/Network/BitTorrent/PeerWire/Handshake.hs b/src/Network/BitTorrent/PeerWire/Handshake.hs
index c4acf5cc..6f4598ae 100644
--- a/src/Network/BitTorrent/PeerWire/Handshake.hs
+++ b/src/Network/BitTorrent/PeerWire/Handshake.hs
@@ -27,10 +27,10 @@ import           Data.ByteString (ByteString)
 import qualified Data.ByteString as B
 import qualified Data.ByteString.Char8 as BC
 import Data.Serialize as S
-import Data.Torrent.InfoHash
 import Network
 import Network.Socket.ByteString
 
+import Data.Torrent
 import Network.BitTorrent.Extension
 import Network.BitTorrent.Peer.ID
 import Network.BitTorrent.Peer.ClientInfo
diff --git a/src/Network/BitTorrent/Tracker/Scrape.hs b/src/Network/BitTorrent/Tracker/Scrape.hs
index c3f4ee64..0181cf9f 100644
--- a/src/Network/BitTorrent/Tracker/Scrape.hs
+++ b/src/Network/BitTorrent/Tracker/Scrape.hs
@@ -28,10 +28,12 @@ import qualified Data.ByteString.Char8 as BC
 import           Data.Map (Map)
 import qualified Data.Map as M
 import Data.Monoid
-import Data.Torrent.InfoHash
 import Network.URI
 import Network.HTTP
 
+import Data.Torrent
+
+
 -- | Information about particular torrent.
 data ScrapeInfo = ScrapeInfo {
     siComplete   :: Int