1 files changed, 4 insertions, 2 deletions

diff --git a/src/Network/Tox.hs b/src/Network/Tox.hs
index 68714224..41deff42 100644
--- a/src/Network/Tox.hs
+++ b/src/Network/Tox.hs
@@ -449,7 +449,9 @@ newTox keydb addr mbSessionsState suppliedDHTKey = do
     dhtclient <- newClient drg dhtnet0 DHT.classify (myAddr tbl4 tbl6) (DHT.handlers crypto . mkrouting) id
                     $ \client net -> onInbound (DHT.updateRouting client (mkrouting client) orouter) net
 
-    orouter <- forkRouteBuilder orouter $ \nid ni -> fmap (\(_,ns,_)->ns) <$> DHT.getNodes dhtclient nid ni
+    orouter' <- forkRouteBuilder orouter
+        $ \nid ni -> fmap (\(_,ns,_)->ns)
+                     <$> DHT.getNodes dhtclient (DHT.nodesOfInterest $ mkrouting dhtclient) nid ni
 
     toks <- do
         nil <- nullSessionTokens
@@ -459,7 +461,7 @@ newTox keydb addr mbSessionsState suppliedDHTKey = do
     onionclient <- newClient oniondrg onionnet (const Onion.classify)
                     (getOnionAlias crypto $ R.thisNode <$> readTVar (DHT.routing4 $ mkrouting dhtclient))
                     (const $ Onion.handlers onionnet (mkrouting dhtclient) toks keydb)
-                    (hookQueries orouter DHT.transactionKey)
+                    (hookQueries orouter' DHT.transactionKey)
                     (const id)
 
     return Tox