1 files changed, 5 insertions, 2 deletions

diff --git a/src/Network/BitTorrent/DHT.hs b/src/Network/BitTorrent/DHT.hs
index aaa1cf33..ab948a2d 100644
--- a/src/Network/BitTorrent/DHT.hs
+++ b/src/Network/BitTorrent/DHT.hs
@@ -17,6 +17,7 @@
 {-# LANGUAGE FlexibleInstances #-}
 {-# LANGUAGE TemplateHaskell   #-}
 {-# LANGUAGE TypeOperators     #-}
+{-# LANGUAGE ScopedTypeVariables #-}
 module Network.BitTorrent.DHT
        ( -- * Distributed Hash Table
          DHT
@@ -70,6 +71,7 @@ import Network.BitTorrent.DHT.Session
 import Network.BitTorrent.DHT.Routing as T hiding (null)
 import qualified Data.Text as Text
 import Data.Monoid
+import Network.KRPC.Message (KMessageOf)
 
 
 {-----------------------------------------------------------------------
@@ -166,7 +168,7 @@ resolveHostName NodeAddr {..} = do
 --
 --   This operation do block, use
 --   'Control.Concurrent.Async.Lifted.async' if needed.
-bootstrap :: Address ip => Maybe BS.ByteString -> [NodeAddr ip] -> DHT ip ()
+bootstrap :: forall ip. Address ip => Maybe BS.ByteString -> [NodeAddr ip] -> DHT ip ()
 bootstrap mbs startNodes = do
   restored <-
     case decode <$> mbs of
@@ -178,7 +180,8 @@ bootstrap mbs startNodes = do
   $(logInfoS) "bootstrap" "Start node bootstrapping"
   let searchAll aliveNodes = do
         nid <- myNodeIdAccordingTo (error "FIXME")
-        C.sourceList [aliveNodes] $= search nid (findNodeQ nid) $$ C.consume
+        nss <- C.sourceList [aliveNodes] $= search nid (findNodeQ nid) $$ C.consume
+        return ( nss :: [[NodeInfo KMessageOf ip ()]] )
   input_nodes <- (restored ++) . T.toList <$> getTable
   -- Step 1: Use iterative searches to flesh out the table..
   do let knowns = map (map $ nodeAddr . fst) input_nodes