1 files changed, 2 insertions, 5 deletions
diff --git a/tests/test_base.py b/tests/test_base.py
index 0788f82..e0a6547 100644
--- a/tests/test_base.py
+++ b/tests/test_base.py
@@ -20,16 +20,13 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 # THE SOFTWARE.
 
-try:
-    import unittest2 as unittest
-except ImportError:
-    import unittest
-
 import warnings
 
 from factory import base
 from factory import declarations
 
+from .compat import unittest
+
 class TestObject(object):
     def __init__(self, one=None, two=None, three=None, four=None):
         self.one = one