Use PieceSize then appropriate

2 files changed, 3 insertions, 2 deletions

diff --git a/src/Network/BitTorrent/Exchange/Message.hs b/src/Network/BitTorrent/Exchange/Message.hs
index 7748f843..d8873f95 100644
--- a/src/Network/BitTorrent/Exchange/Message.hs
+++ b/src/Network/BitTorrent/Exchange/Message.hs
@@ -864,7 +864,7 @@ instance PeerMessage ExtendedMetadata where
 
 -- | All 'Piece's in 'MetadataData' messages MUST have size equal to
 -- this value. The last trailing piece can be shorter.
-metadataPieceSize :: Int
+metadataPieceSize :: P.PieceSize
 metadataPieceSize = 16 * 1024
 
 isLastPiece :: P.Piece a -> Int -> Bool
diff --git a/src/Network/BitTorrent/Exchange/Session/Metadata.hs b/src/Network/BitTorrent/Exchange/Session/Metadata.hs
index 7e14f493..174550fa 100644
--- a/src/Network/BitTorrent/Exchange/Session/Metadata.hs
+++ b/src/Network/BitTorrent/Exchange/Session/Metadata.hs
@@ -38,7 +38,8 @@ data Status = Status
 
 makeLenses ''Status
 
-nullStatus :: PieceSize -> Status
+-- | Create a new scheduler for infodict of the given size.
+nullStatus :: Int -> Status
 nullStatus ps = Status [] (Block.empty ps)
 
 type Updates a = State Status a