1 files changed, 5 insertions, 5 deletions

diff --git a/dht/src/Network/QueryResponse.hs b/dht/src/Network/QueryResponse.hs
index 8e32899f..1bfa995f 100644
--- a/dht/src/Network/QueryResponse.hs
+++ b/dht/src/Network/QueryResponse.hs
@@ -122,9 +122,9 @@ layerTransport parse encode tr =
 -- is used to share the same underlying socket, so be sure to fork a thread for
 -- both returned 'Transport's to avoid hanging.
 partitionTransportM :: ((b,a) -> IO (Either (x,xaddr) (b,a)))
-                      -> ((x,xaddr) -> IO (Maybe (c,a)))
+                      -> ((y,xaddr) -> IO (Maybe (c,a)))
                       -> TransportA err a b c
-                      -> IO (Transport err xaddr x, TransportA err a b c)
+                      -> IO (TransportA err xaddr x y, TransportA err a b c)
 partitionTransportM parse encodex tr = do
     tchan <- atomically newTChan
     let ytr = tr { awaitMessage = \kont -> fix $ \again -> do
@@ -151,9 +151,9 @@ partitionTransportM parse encodex tr = do
 -- is used to share the same underlying socket, so be sure to fork a thread for
 -- both returned 'Transport's to avoid hanging.
 partitionTransport :: ((b,a) -> Either (x,xaddr) (b,a))
-                      -> ((x,xaddr) -> Maybe (b,a))
-                      -> Transport err a b
-                      -> IO (Transport err xaddr x, Transport err a b)
+                      -> ((y,xaddr) -> Maybe (c,a))
+                      -> TransportA err a b c
+                      -> IO (TransportA err xaddr x y, TransportA err a b c)
 partitionTransport parse encodex tr =
     partitionTransportM (return . parse) (return . encodex) tr