diff --git a/tests/test_phase_base.py b/tests/test_phase_base.py
index d0588fea..a8b9469a 100644
--- a/tests/test_phase_base.py
+++ b/tests/test_phase_base.py
@@ -2,7 +2,10 @@
 # -*- coding: utf-8 -*-
 
 import mock
-import unittest
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 import os
 import sys