diff --git a/tests/restful/RestfulClientTest.py b/tests/restful/RestfulClientTest.py
index 5d3421a7..06aef8c6 100644
--- a/tests/restful/RestfulClientTest.py
+++ b/tests/restful/RestfulClientTest.py
@@ -12,7 +12,8 @@ class TestRestfulClient(RestfulClient):
     
     def __init__(self):
         urlBase = 'https://httpbin.org'
-        super(TestRestfulClient, self).__init__(urlBase, requestsSessionProvider)
+        super(TestRestfulClient, self).__init__()
+        self.init(urlBase)
         
         self.p = Promise()
     
diff --git a/tests/restful/WebSocketClientTest.py b/tests/restful/WebSocketClientTest.py
index 5006265c..79b47510 100644
--- a/tests/restful/WebSocketClientTest.py
+++ b/tests/restful/WebSocketClientTest.py
@@ -9,8 +9,9 @@ from vnpy.restful.WebSocketClient import WebsocketClient
 class TestWebsocketClient(WebsocketClient):
     
     def __init__(self):
-        urlBase = 'wss://echo.websocket.org'
-        super(TestWebsocketClient, self).__init__(urlBase)
+        host = 'wss://echo.websocket.org'
+        super(TestWebsocketClient, self).__init__()
+        self.init(host)
         self.p = Promise()
     
     def onMessage(self, packet):