1 files changed, 5 insertions, 4 deletions

diff --git a/dht/src/Network/Tox.hs b/dht/src/Network/Tox.hs
index 6b39d57a..3dd1d48e 100644
--- a/dht/src/Network/Tox.hs
+++ b/dht/src/Network/Tox.hs
@@ -245,14 +245,15 @@ isLocalHost _                       = False
 
 addVerbosity :: Transport err SockAddr ByteString -> Transport err SockAddr ByteString
 addVerbosity tr =
-    tr { awaitMessage = \kont -> awaitMessage tr $ \m -> do
+    tr { awaitMessage = do
+            (m,io) <- awaitMessage tr
             case m of
-              Arrival addr msg -> do
+              Arrival addr msg -> return $ (,) m $ do
+                io
                 when (not (B.null msg || elem (B.head msg) [0,1,2,4,0x81,0x82,0x8c,0x8d])) $ do
                     mapM_ (\x -> dput XMisc ( (show addr) ++ " --> " ++ x))
                           $ xxd 0 msg
-              _ -> return ()
-            kont m
+              _ -> return (m,io)
        , sendMessage = \addr msg -> do
             when (not (B.null msg || elem (B.head msg) [0,1,2,4,0x81,0x8c,0x8d])) $ do
                 mapM_ (\x -> dput XMisc ( (show addr) ++ " <-- " ++ x))