diff --git a/aiogremlin/remote/remote_connection.py b/aiogremlin/remote/remote_connection.py
index 9cc8f6583128cfb9bbb5f7899b2c334d30610de7..104f94a6b410b94578c648f2b5b9e49223647640 100644
--- a/aiogremlin/remote/remote_connection.py
+++ b/aiogremlin/remote/remote_connection.py
@@ -1,9 +1,7 @@
-from gremlin_python.process import traversal
+from gremlin_python.driver.remote_connection import RemoteStrategy
 
 
-class AsyncRemoteStrategy(traversal.TraversalStrategy):
-    def __init__(self, remote_connection):
-        self.remote_connection = remote_connection
+class AsyncRemoteStrategy(RemoteStrategy):
 
     async def apply(self, traversal):
         if traversal.traversers is None: