diff --git a/tests/test_base.py b/tests/test_base.py
index 0543744..7d34878 100755
--- a/tests/test_base.py
+++ b/tests/test_base.py
@@ -4,7 +4,11 @@
 
 """Test the various functions from 'base'."""
 
-import unittest
+try:  # pragma: no cover
+    import unittest2 as unittest
+except ImportError:  # pragma: no cover
+    import unittest
+
 
 from semantic_version import base