1 files changed, 9 insertions, 9 deletions
diff --git a/test/test_package.py b/test/test_package.py
index e520f3f..aa42c9c 100644
--- a/test/test_package.py
+++ b/test/test_package.py
@@ -6,20 +6,20 @@ class TestPackage:
         assert kafka1.codec.__name__ == "kafka.codec"
 
     def test_submodule_namespace(self):
-        import kafka.client as client1
-        assert client1.__name__ == "kafka.client"
+        import kafka.client_async as client1
+        assert client1.__name__ == "kafka.client_async"
 
-        from kafka import client as client2
-        assert client2.__name__ == "kafka.client"
+        from kafka import client_async as client2
+        assert client2.__name__ == "kafka.client_async"
 
-        from kafka.client import SimpleClient as SimpleClient1
-        assert SimpleClient1.__name__ == "SimpleClient"
+        from kafka.client_async import KafkaClient as KafkaClient1
+        assert KafkaClient1.__name__ == "KafkaClient"
+
+        from kafka import KafkaClient as KafkaClient2
+        assert KafkaClient2.__name__ == "KafkaClient"
 
         from kafka.codec import gzip_encode as gzip_encode1
         assert gzip_encode1.__name__ == "gzip_encode"
 
-        from kafka import SimpleClient as SimpleClient2
-        assert SimpleClient2.__name__ == "SimpleClient"
-
         from kafka.codec import snappy_encode
         assert snappy_encode.__name__ == "snappy_encode"