1 files changed, 2 insertions, 2 deletions

diff --git a/src/Network/BitTorrent/Exchange/Protocol.hs b/src/Network/BitTorrent/Exchange/Protocol.hs
index 83774f06..a285f8d2 100644
--- a/src/Network/BitTorrent/Exchange/Protocol.hs
+++ b/src/Network/BitTorrent/Exchange/Protocol.hs
@@ -435,7 +435,7 @@ instance Serialize Message where
 
 instance Binary Message where
   get = do
-    len <- undefined --getInt
+    len <- getIntB
 --    _   <- lookAhead $ ensure len
     if len == 0 then return KeepAlive
       else do
@@ -463,7 +463,7 @@ instance Binary Message where
 
     where
       getBlock :: Int -> B.Get Block
-      getBlock len = Block <$> getIntB <*> getIntB <*> B.getBytes len
+      getBlock len = Block <$> getIntB <*> getIntB <*> B.getByteString len
       {-# INLINE getBlock #-}
 
   put KeepAlive     = putIntB 0