3 files changed, 14 insertions, 4 deletions

diff --git a/bench/Main.hs b/bench/Main.hs
index 97f97425..13727ff9 100644
--- a/bench/Main.hs
+++ b/bench/Main.hs
@@ -3,6 +3,7 @@
 {-# OPTIONS_GHC -fno-warn-orphans  #-}
 module Main (main) where
 import Control.Monad
+import Control.Monad.Logger
 import Control.Monad.Reader
 import Criterion.Main
 import Data.ByteString as BS
@@ -11,6 +12,9 @@ import Network.KRPC
 instance KRPC ByteString ByteString where
   method = "echo"
 
+instance MonadLogger IO where
+  monadLoggerLog _ _ _ _ = return ()
+
 echo :: Handler IO
 echo = handler $ \ _ bs -> return (bs :: ByteString)
 
@@ -26,7 +30,7 @@ main = withManager addr [echo] $ \ m -> (`runReaderT` m) $ do
     repetitions  = [1, 10, 100, 1000]
     benchmarks m = [mkbench m r s | r <- repetitions, s <- sizes]
       where
-        mkbench m r n =
+        mkbench action r n =
           bench (show r ++ "times" ++ "/" ++ show n ++ "bytes") $ nfIO $
             replicateM r $
-              runReaderT (query addr (BS.replicate n 0)) m
+              runReaderT (query addr (BS.replicate n 0)) action
diff --git a/krpc.cabal b/krpc.cabal
index 7b0cafa2..b5004026 100644
--- a/krpc.cabal
+++ b/krpc.cabal
@@ -70,6 +70,7 @@ test-suite spec
                      , bytestring
                      , network
                      , mtl
+                     , monad-logger
                      , hspec
                      , QuickCheck
                      , quickcheck-instances
@@ -84,6 +85,7 @@ benchmark bench
   build-depends:       base == 4.*
                      , bytestring
                      , mtl
+                     , monad-logger
                      , criterion
                      , krpc
   ghc-options:         -O2 -fforce-recomp
\ No newline at end of file
diff --git a/tests/Network/KRPCSpec.hs b/tests/Network/KRPCSpec.hs
index 27148682..7f5b2794 100644
--- a/tests/Network/KRPCSpec.hs
+++ b/tests/Network/KRPCSpec.hs
@@ -1,7 +1,8 @@
-{-# LANGUAGE OverloadedStrings #-}
+{-# LANGUAGE OverloadedStrings    #-}
+{-# OPTIONS_GHC -fno-warn-orphans #-}
 module Network.KRPCSpec (spec) where
+import Control.Monad.Logger
 import Control.Monad.Reader
-import Network.Socket (SockAddr (..))
 import Network.KRPC
 import Network.KRPC.MethodSpec hiding (spec)
 import Test.Hspec
@@ -16,6 +17,9 @@ handlers =
  , handler $ \ _ (Echo a) -> return (Echo (a :: Int))
  ]
 
+instance MonadLogger IO where
+  monadLoggerLog _ _ _ _ = return ()
+
 spec :: Spec
 spec = do
   describe "query" $ do