1 files changed, 17 insertions, 16 deletions

diff --git a/src/Network/SessionTransports.hs b/src/Network/SessionTransports.hs
index 17763e4e..e9daf6c1 100644
--- a/src/Network/SessionTransports.hs
+++ b/src/Network/SessionTransports.hs
@@ -49,7 +49,7 @@ newSession :: Sessions raw
                     -> (addr -> y -> IO raw)
                     -> (SockAddr -> raw -> IO (Maybe (x, addr)))
                     -> SockAddr
-                    -> IO (Maybe (TransportA err addr x y))
+                    -> IO (Maybe (Int,TransportA err addr x y))
 newSession Sessions{sessionsByAddr,sessionsById,sessionIds,sessionsSendRaw} unwrap wrap addr0 = do
     mvar <- newEmptyMVar
     let saddr = -- Canonical in case of 6-mapped-4 addresses.
@@ -69,21 +69,22 @@ newSession Sessions{sessionsByAddr,sessionsById,sessionIds,sessionsSendRaw} unwr
                                        $ IntMap.singleton sid handlePacket
             return sid
     forM msid $ \sid -> do
-    return Transport
-        { awaitMessage = \kont -> do
-            x <- takeMVar mvar
-            kont $! Right <$> x
-        , sendMessage = \addr x -> do
-            x' <- unwrap addr x
-            sessionsSendRaw saddr x'
-        , closeTransport = do
-            tryTakeMVar mvar
-            putMVar mvar Nothing
-            atomically $ do
-                modifyTVar' sessionIds $ S.delete sid
-                modifyTVar' sessionsById $ IntMap.delete sid
-                modifyTVar' sessionsByAddr $ Map.alter (rmSession sid) saddr
-        }
+    let tr = Transport
+            { awaitMessage = \kont -> do
+                x <- takeMVar mvar
+                kont $! Right <$> x
+            , sendMessage = \addr x -> do
+                x' <- unwrap addr x
+                sessionsSendRaw saddr x'
+            , closeTransport = do
+                tryTakeMVar mvar
+                putMVar mvar Nothing
+                atomically $ do
+                    modifyTVar' sessionIds $ S.delete sid
+                    modifyTVar' sessionsById $ IntMap.delete sid
+                    modifyTVar' sessionsByAddr $ Map.alter (rmSession sid) saddr
+            }
+    return (sid,tr)
 
 sessionHandler :: Sessions x -> (SockAddr -> x -> IO (Maybe (x -> x)))
 sessionHandler Sessions{sessionsByAddr} = \addr0 x -> do