~ Use PortNumber instead of Word16 in Port message.

1 files changed, 5 insertions, 0 deletions

diff --git a/bench/Main.hs b/bench/Main.hs
index 269e60f5..6e8a0ce3 100644
--- a/bench/Main.hs
+++ b/bench/Main.hs
@@ -7,10 +7,15 @@ import Criterion.Main
 import Data.ByteString (ByteString)
 import qualified Data.ByteString as B
 import Data.Serialize
+import Network
+
 import Network.BitTorrent as BT
 import Data.Bitfield as BT
 
 
+instance NFData PortNumber where
+  rnf = rnf . (fromIntegral :: PortNumber -> Int)
+
 instance NFData BlockIx where
   rnf (BlockIx a b c) = a `deepseq` b `deepseq` rnf c