1 files changed, 6 insertions, 2 deletions

diff --git a/Presence/XMPP.hs b/Presence/XMPP.hs
index c7525159..4db055ea 100644
--- a/Presence/XMPP.hs
+++ b/Presence/XMPP.hs
@@ -922,12 +922,14 @@ fromPeer sock session = doNestingXML $ do
     withJust (elementAttrs "stream" xml) $ \stream_attrs -> do
     log $ "stream atributes: " <++> bshow stream_attrs
 
+    let doTimeout = Thunk (closePeerSession session)
+
     fix $ \loop -> do
         log "waiting for stanza."
         whenJust nextElement $ \stanza -> do
             stanza_lvl <- nesting
 
-            liftIO $ sendPeerMessage session (ActivityBump sock) -- reset ping timer
+            liftIO $ sendPeerMessage session (ActivityBump doTimeout) -- reset ping timer
 
             let unhandledStanza = do
                     xs <- gatherElement stanza Seq.empty
@@ -1181,7 +1183,9 @@ toPeer sock cache chan fail = do
                             remote <- liftIO $ getPeerName sock
                             liftIO . debugL $ "PING TIMEOUT: " <++> showPeer (RemotePeer remote)
                             fromsock <- liftIO $ atomically $ readTMVar sockref
-                            liftIO $ sClose fromsock
+                            -- liftIO $ sClose fromsock
+                            liftIO $ runThunk fromsock
+
                             return () -- PING TIMEOUT (loop quits)
                     x -> error ("What? "++show x)
               where makePing = do