diff --git a/django_mailbox/tests/test_integration_imap.py b/django_mailbox/tests/test_integration_imap.py
index 437cbe0..1b17654 100644
--- a/django_mailbox/tests/test_integration_imap.py
+++ b/django_mailbox/tests/test_integration_imap.py
@@ -1,5 +1,5 @@
 import os
-from unittest import SkipTest
+from unittest2 import SkipTest
 import uuid
 
 from six.moves.urllib import parse
diff --git a/setup.py b/setup.py
index 3023819..8db714b 100755
--- a/setup.py
+++ b/setup.py
@@ -5,6 +5,7 @@ from django_mailbox import __version__ as version_string
 tests_require = [
     'django',
     'mock',
+    'unittest2',
 ]
 
 gmail_oauth2_require = [