2 files changed, 4 insertions, 2 deletions

diff --git a/bench/Main.hs b/bench/Main.hs
index fdf76cc2..024d4d93 100644
--- a/bench/Main.hs
+++ b/bench/Main.hs
@@ -6,10 +6,11 @@ import Data.ByteString (ByteString)
 import qualified Data.ByteString as B
 import Criterion.Main
 import Network.KRPC
+import Network.Socket
 
 
 addr :: RemoteAddr
-addr = (0, 6000)
+addr = SockAddrInet 6000 0
 
 echo :: Method ByteString ByteString
 echo = method "echo" ["x"] ["x"]
diff --git a/bench/Server.hs b/bench/Server.hs
index 444362c1..ef20c08a 100644
--- a/bench/Server.hs
+++ b/bench/Server.hs
@@ -3,10 +3,11 @@ module Main (main) where
 
 import Data.ByteString (ByteString)
 import Network.KRPC
+import Network.Socket
 
 
 echo :: Method ByteString ByteString
 echo = method "echo" ["x"] ["x"]
 
 main :: IO ()
-main = server 6000 [ echo ==> return ]
+main = server (SockAddrInet 6000 0) [ echo ==> return ]