~ Use safe spawnP2P.

1 files changed, 1 insertions, 4 deletions

diff --git a/src/Network/BitTorrent/Exchange.hs b/src/Network/BitTorrent/Exchange.hs
index 5be9ae73..a0e7df8c 100644
--- a/src/Network/BitTorrent/Exchange.hs
+++ b/src/Network/BitTorrent/Exchange.hs
@@ -171,10 +171,7 @@ runP2P se addr p2p = waitVacancy se $ runSession se addr p2p
 -- some other loop. Note that this function may block while waiting
 -- for a vacant place: use forkIO and runP2P instead.
 spawnP2P :: SwarmSession -> PeerAddr -> P2P () -> IO ThreadId
-spawnP2P se addr p2p = do
-  enterSwarm se
-  forkIO $ do
-    runSession se addr p2p `finally` leaveSwarm se
+spawnP2P se addr p2p = forkThrottle se $ runSession se addr p2p
 
 -- TODO unify this all using PeerConnection
 {-