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