diff --git a/tests/__init__.py b/tests/__init__.py
index e69de29b..b4d38ccc 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -0,0 +1,7 @@
+import sys
+
+if sys.version_info >= (2,7):
+    import unittest
+else:
+    import unittest2 as unittest
+
diff --git a/tests/testcases.py b/tests/testcases.py
index c1da5389..4ac61d9d 100644
--- a/tests/testcases.py
+++ b/tests/testcases.py
@@ -1,10 +1,10 @@
 from docker import Client
 from fig.service import Service
 from fig.cli.utils import docker_url
-from unittest2 import TestCase
+from . import unittest
 
 
-class DockerClientTestCase(TestCase):
+class DockerClientTestCase(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
         cls.client = Client(docker_url())