diff --git a/src/test/setup/cord-test.py b/src/test/setup/cord-test.py
index f2c4600..8b7e0fc 100755
--- a/src/test/setup/cord-test.py
+++ b/src/test/setup/cord-test.py
@@ -31,7 +31,10 @@
 from CordTestServer import cord_test_server_start,cord_test_server_stop,cord_test_server_shutdown,CORD_TEST_HOST,CORD_TEST_PORT
 from TestManifest import TestManifest
 from VolthaCtrl import VolthaService
-from docker import Client
+try:
+    from docker import APIClient as Client
+except:
+    from docker import Client
 from docker.utils import kwargs_from_env
 from Xos import XosServiceProfile
 try:
diff --git a/src/test/setup/requirements.txt b/src/test/setup/requirements.txt
index 94a195c..57cdaed 100644
--- a/src/test/setup/requirements.txt
+++ b/src/test/setup/requirements.txt
@@ -2,7 +2,7 @@
 scapy==2.3.2
 monotonic
 configObj
-docker-py
+docker
 nsenter
 pyroute2
 netaddr