diff --git a/test_elasticsearch/test_server/__init__.py b/test_elasticsearch/test_server/__init__.py
index 53adbe0f..9c116675 100644
--- a/test_elasticsearch/test_server/__init__.py
+++ b/test_elasticsearch/test_server/__init__.py
@@ -1,7 +1,5 @@
-from elasticsearch.helpers.test import (
-    get_test_client,
-    ElasticsearchTestCase as BaseTestCase,
-)
+from elasticsearch.helpers import test
+from elasticsearch.helpers.test import ElasticsearchTestCase as BaseTestCase
 
 client = None
 
@@ -18,7 +16,7 @@ def get_client(**kwargs):
         new_client = local_get_client(**kwargs)
     except ImportError:
         # fallback to using vanilla client
-        new_client = get_test_client(**kwargs)
+        new_client = test.get_test_client(**kwargs)
 
     if not kwargs:
         client = new_client