2 files changed, 3 insertions, 3 deletions

diff --git a/tests/Network/BitTorrent/DHT/QuerySpec.hs b/tests/Network/BitTorrent/DHT/QuerySpec.hs
index b89ce4ff..33114b40 100644
--- a/tests/Network/BitTorrent/DHT/QuerySpec.hs
+++ b/tests/Network/BitTorrent/DHT/QuerySpec.hs
@@ -28,7 +28,7 @@ nullLogger _ _ _ _ = return ()
 
 simpleDHT :: [NodeHandler IPv4] -> DHT IPv4 a -> IO a
 simpleDHT hs m =
-  bracket (startNode hs def myAddr nullLogger) stopNode $ \ node ->
+  bracket (newNode hs def myAddr nullLogger) closeNode $ \ node ->
     runDHT node m
 
 getBootInfo :: IO (NodeInfo IPv4)
diff --git a/tests/Network/BitTorrent/DHT/SessionSpec.hs b/tests/Network/BitTorrent/DHT/SessionSpec.hs
index 41852763..b3b9a9fc 100644
--- a/tests/Network/BitTorrent/DHT/SessionSpec.hs
+++ b/tests/Network/BitTorrent/DHT/SessionSpec.hs
@@ -39,10 +39,10 @@ spec :: Spec
 spec = do
   describe "session" $ do
     it "is active until stopNode called" $ do
-      node <- startNode [] def myAddr nullLogger
+      node <- newNode [] def myAddr nullLogger
       runDHT node monadActive `shouldReturn` True
       runDHT node monadActive `shouldReturn` True
-      stopNode node
+      closeNode node
       runDHT node monadActive `shouldReturn` False
 
   describe "tokens" $ do