diff --git a/requirements.txt b/requirements.txt
index 4eecc7c..073dd21 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,7 @@
 pillow>=7.1.2
 threadpoolctl
-scikit-learn
 numpy>=1.10.2
 scipy
 scikit-image
 scikit-learn
+future
diff --git a/setup.py b/setup.py
index ef0def0..369d7db 100755
--- a/setup.py
+++ b/setup.py
@@ -15,7 +15,6 @@
             "scikit-image",
             "scikit-learn",
             "future",
-            "unittest2",
             "threadpoolctl"
       ],
       keywords=['deep learning', 'image segmentation', 'image classification', 'medical image analysis',
diff --git a/tests/test_axis_mirroring.py b/tests/test_axis_mirroring.py
index 1b4598c..19465e2 100644
--- a/tests/test_axis_mirroring.py
+++ b/tests/test_axis_mirroring.py
@@ -12,8 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import unittest
-
-import unittest2
 import numpy as np
 from skimage import data
 
@@ -22,7 +20,7 @@
 from batchgenerators.transforms.spatial_transforms import MirrorTransform
 
 
-class TestMirrorAxis(unittest2.TestCase):
+class TestMirrorAxis(unittest.TestCase):
     def setUp(self):
         self.seed = 1234