2 files changed, 4 insertions, 3 deletions

diff --git a/src/Network/BitTorrent/DHT/Query.hs b/src/Network/BitTorrent/DHT/Query.hs
index ac53bd91..cb7d5c5f 100644
--- a/src/Network/BitTorrent/DHT/Query.hs
+++ b/src/Network/BitTorrent/DHT/Query.hs
@@ -160,7 +160,7 @@ search k action = do
     $(logWarnS) "search" "start query"
     responses <- lift $ queryParallel (action <$> batch)
     let (nodes, results) = partitionEithers responses
-    $(logWarnS) "search" "done query"
+    $(logWarnS) "search" ("done query more:" <> T.pack (show $ L.length nodes))
     leftover $ L.concat nodes
     mapM_ yield results
 
diff --git a/src/Network/BitTorrent/DHT/Session.hs b/src/Network/BitTorrent/DHT/Session.hs
index 9aa25866..38b3ed11 100644
--- a/src/Network/BitTorrent/DHT/Session.hs
+++ b/src/Network/BitTorrent/DHT/Session.hs
@@ -299,8 +299,9 @@ newNode :: Address ip
           -> Options          -- ^ various dht options;
           -> NodeAddr ip      -- ^ node address to bind;
           -> LogFun           -- ^
+          -> Maybe NodeId     -- ^ use this NodeId, if not given a new one is generated.
           -> IO (Node ip)     -- ^ a new DHT node running at given address.
-newNode hs opts naddr logger = do
+newNode hs opts naddr logger mbid = do
     s <- createInternalState
     runInternalState initNode s
       `onException` closeInternalState s
@@ -311,7 +312,7 @@ newNode hs opts naddr logger = do
       s      <- getInternalState
       (_, m) <- allocate (newManager rpcOpts nodeAddr hs) closeManager
       liftIO $ do
-        myId   <- genNodeId
+        myId   <- maybe genNodeId return mbid
         node   <- Node opts myId s m
                 <$> newMVar (nullTable myId (optBucketCount opts))
                 <*> newTVarIO def