--- a/docs/sample.py
+++ b/docs/sample.py
@@ -1,4 +1,4 @@
-import unittest2
+import unittest
 
 import deprecation
 
@@ -70,7 +70,7 @@
     return 1
 
 
-class Tests(unittest2.TestCase):
+class Tests(unittest.TestCase):
 
     @deprecation.fail_if_not_removed
     def test_won(self):
--- a/tests/test_deprecation.py
+++ b/tests/test_deprecation.py
@@ -12,14 +12,14 @@
 
 # As we unfortunately support Python 2.7, it lacks TestCase.subTest which
 # is in 3.4+ or in unittest2
-import unittest2
+import unittest
 import warnings
 
 import deprecation
 from datetime import date
 
 
-class Test_deprecated(unittest2.TestCase):
+class Test_deprecated(unittest.TestCase):
 
     def test_args_set_on_base_class(self):
         args = (1, 2, 3, 4)
@@ -230,7 +230,7 @@
             self.assertEqual(sot.method(), ret_val)
 
 
-class Test_fail_if_not_removed(unittest2.TestCase):
+class Test_fail_if_not_removed(unittest.TestCase):
 
     @deprecation.deprecated(deprecated_in="1.0", current_version="2.0")
     def _deprecated_method(self):
@@ -259,7 +259,7 @@
         except AssertionError:
             self.fail("A DeprecatedWarning shouldn't cause a failure")
 
-    @unittest2.expectedFailure
+    @unittest.expectedFailure
     @deprecation.fail_if_not_removed
     def test_literal_UnsupportedWarning(self):
         self._unsupported_method()
