diff --git a/contrib/pylightning/lightning/lightning.py b/contrib/pylightning/lightning/lightning.py
index ce3bcec03..829832e36 100644
--- a/contrib/pylightning/lightning/lightning.py
+++ b/contrib/pylightning/lightning/lightning.py
@@ -1,15 +1,14 @@
-from concurrent import futures
-
 import json
 import logging
 import socket
 
 
 class UnixDomainSocketRpc(object):
-    def __init__(self, socket_path, executor=None):
+    def __init__(self, socket_path, executor=None, logger=logging):
         self.socket_path = socket_path
         self.decoder = json.JSONDecoder()
         self.executor = executor
+        self.logger = logger
 
     @staticmethod
     def _writeobj(sock, obj):
@@ -45,7 +44,7 @@ class UnixDomainSocketRpc(object):
         return wrapper
 
     def _call(self, method, args=None):
-        logging.debug("Calling %s with arguments %r", method, args)
+        self.logger.debug("Calling %s with payload %r", method, payload)
 
         sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
         sock.connect(self.socket_path)
@@ -57,12 +56,12 @@ class UnixDomainSocketRpc(object):
         resp = self._readobj(sock)
         sock.close()
 
-        logging.debug("Received response for %s call: %r", method, resp)
         if 'error' in resp:
             raise ValueError("RPC call failed: {}".format(resp['error']))
         elif 'result' not in resp:
             raise ValueError("Malformed response, 'result' missing.")
         return resp['result']
+        self.logger.debug("Received response for %s call: %r", method, resp)
 
 
 class LightningRpc(UnixDomainSocketRpc):