diff --git a/setup.py b/setup.py
index a0559d5..b102124 100755
--- a/setup.py
+++ b/setup.py
@@ -97,6 +97,6 @@ setup(
        'cython>=0.28',
        'setuptools_scm>=1.11.1',
     ] + pytest_runner,
-    tests_require=['unittest2', 'pytest'],
+    tests_require=['pytest'],
     cmdclass=cmdclass,
 )
diff --git a/tests/test_pyclipper.py b/tests/test_pyclipper.py
index c151a85..829b77a 100644
--- a/tests/test_pyclipper.py
+++ b/tests/test_pyclipper.py
@@ -4,7 +4,10 @@ Tests for Pyclipper wrapper library.
 """
 
 from __future__ import print_function
-from unittest2 import TestCase, main
+try:
+  from unittest2 import TestCase, main
+except ImportError:
+  from unittest import TestCase, main
 import sys
 
 if sys.version_info < (3,):