1 files changed, 4 insertions, 2 deletions
diff --git a/test/lib/base.py b/test/lib/base.py
index 10b50b46..8e2c5f22 100644
--- a/test/lib/base.py
+++ b/test/lib/base.py
@@ -17,8 +17,10 @@ from fabric.api import local, lcd
 from fabric import colors
 from fabric.utils import indent
 from fabric.state import env, output
-from docker import Client
-
+try:
+    from docker import Client
+except ImportError:
+    from docker import APIClient as Client
 import netaddr
 import os
 import time