2 files changed, 7 insertions, 8 deletions

diff --git a/tests/Network/BitTorrent/Tracker/ListSpec.hs b/tests/Network/BitTorrent/Tracker/ListSpec.hs
index 03379136..bba9d0e2 100644
--- a/tests/Network/BitTorrent/Tracker/ListSpec.hs
+++ b/tests/Network/BitTorrent/Tracker/ListSpec.hs
@@ -17,7 +17,7 @@ uris = fmap (fromJust . parseURI . renderURI) [1..10 :: Int]
   where
     renderURI n = "http://" ++ show n ++ ".org"
 
-list :: TrackerList URI
+list :: TrackerList ()
 list = trackerList def { tAnnounceList = Just [uris] }
 
 spec :: Spec
@@ -28,14 +28,13 @@ spec = do
       list' `shouldSatisfy` (/= list)
 
     it "traverseAll" $ do
-      xs <- traverseAll (\ uri -> if uri == L.last uris
+      xs <- traverseAll (\ (uri, _) -> if uri == L.last uris
                then throwIO (GenericException "")
-               else return uri { uriScheme = "udp://" }) list
-      let udps = F.sum $ fmap (fromEnum . ("udp://" ==) . uriScheme) xs
-      udps `shouldBe` pred (L.length uris)
+               else return ()) list
+      return ()
 
     it "traverseTiers" $ do
-      xs' <- traverseTiers (\ uri -> if uri == L.last uris then return uri
+      xs' <- traverseTiers (\ (uri, _) -> if uri == L.last uris then return ()
                               else throwIO (GenericException "")) list
 
       return ()
diff --git a/tests/Network/BitTorrent/Tracker/SessionSpec.hs b/tests/Network/BitTorrent/Tracker/SessionSpec.hs
index 2c471752..72936ee7 100644
--- a/tests/Network/BitTorrent/Tracker/SessionSpec.hs
+++ b/tests/Network/BitTorrent/Tracker/SessionSpec.hs
@@ -16,11 +16,11 @@ testSession :: Bool -> (Manager -> Session -> IO ()) -> IO ()
 testSession runEmpty action = do
   t <- getTestTorrent
   withManager def def $ \ m -> do
-    withSession (idInfoHash (tInfoDict t)) (trackerList t) $ \ s ->
+    withSession m (idInfoHash (tInfoDict t)) (trackerList t) $ \ s ->
       action m s
 
     when runEmpty $ do
-      withSession (idInfoHash (tInfoDict t)) def $ \ s ->
+      withSession m (idInfoHash (tInfoDict t)) def $ \ s ->
         action m s
 
 spec :: Spec