1 files changed, 3 insertions, 3 deletions

diff --git a/src/Network/Tox/Crypto/Handlers.hs b/src/Network/Tox/Crypto/Handlers.hs
index c5476371..089a991d 100644
--- a/src/Network/Tox/Crypto/Handlers.hs
+++ b/src/Network/Tox/Crypto/Handlers.hs
@@ -28,7 +28,7 @@ data NetCryptoSessionStatus = Unaccepted | Accepted | Confirmed
 
 
 type IOHook addr x = addr -> x -> IO (Maybe (x -> x))
-type NetCryptoHook = IOHook SockAddr CryptoData
+type NetCryptoHook = IOHook NetCryptoSession CryptoData
 
 
 data NetCryptoSession = NCrypto { ncState         :: TVar NetCryptoSessionStatus
@@ -149,7 +149,7 @@ cryptoNetHandler sessions addr (NetCrypto (CryptoPacket nonce16 encrypted)) = do
     -- Handle Encrypted Message
     case Map.lookup addr sessionsmap of
         Nothing -> return Nothing -- drop packet, we have no session
-        Just (NCrypto {ncState, ncHooks,ncSessionSecret,ncTheirSessionPublic,ncTheirBaseNonce}) -> do
+        Just session@(NCrypto {ncState, ncHooks,ncSessionSecret,ncTheirSessionPublic,ncTheirBaseNonce}) -> do
             theirBaseNonce <- atomically $ readTVar ncTheirBaseNonce
             -- Try to decrypt message
             let diff :: Word16
@@ -187,7 +187,7 @@ cryptoNetHandler sessions addr (NetCrypto (CryptoPacket nonce16 encrypted)) = do
                                                 case (hooks,cd) of
                                                     ([],_) ->  return Nothing
                                                     (hook:more,cd) -> do
-                                                        r <- hook addr cd :: IO (Maybe (CryptoData -> CryptoData))
+                                                        r <- hook session cd :: IO (Maybe (CryptoData -> CryptoData))
                                                         case r of
                                                             Just f -> let newcd = f cd
                                                                           newtyp = newcd ^. messageType