1 files changed, 3 insertions, 3 deletions

diff --git a/tests/Network/BitTorrent/DHT/QuerySpec.hs b/tests/Network/BitTorrent/DHT/QuerySpec.hs
index 6b3b5485..93f78263 100644
--- a/tests/Network/BitTorrent/DHT/QuerySpec.hs
+++ b/tests/Network/BitTorrent/DHT/QuerySpec.hs
@@ -34,7 +34,7 @@ simpleDHT hs m =
 getBootInfo :: IO (NodeInfo IPv4)
 getBootInfo = do
   startAddr <- resolveHostName (L.head defaultBootstrapNodes)
-  simpleDHT [] $ pingQ startAddr
+  simpleDHT [] $ fmap fst (pingQ startAddr)
 
 spec :: Spec
 spec = parallel $ do
@@ -55,14 +55,14 @@ spec = parallel $ do
     it "findNode" $ do
       startInfo <- getBootInfo
       _ <- simpleDHT [] $ do
-        nid <- asks thisNodeId
+        nid <- myNodeIdAccordingTo (read "8.8.8.8:6881")
         findNodeQ nid startInfo
       return ()
 
     it "getPeers" $ do
       startInfo <- getBootInfo
       peers <- simpleDHT [] $ do
-        nid <- asks thisNodeId
+        nid <- myNodeIdAccordingTo (read "8.8.8.8:6881")
 
         -- we should not run getPeers query on boot node, because
         -- it may not support it