diff --git a/setup.py b/setup.py
index b8197cc..617c1cc 100644
--- a/setup.py
+++ b/setup.py
@@ -5,11 +5,12 @@ VERSION = "0.27.0"
 NAME="websocket_client"
 
 install_requires = ["six"]
+tests_require = []
 if sys.version_info[0] == 2:
     if sys.version_info[1] < 7 or (sys.version_info[1] == 7 and sys.version_info[2]< 9):
         install_requires.append('backports.ssl_match_hostname')
     if sys.version_info[1] < 7:
-        install_requires.append('unittest2==0.8.0')
+        tests_require.append('unittest2==0.8.0')
         install_requires.append('argparse')
 
 setup(
@@ -42,5 +43,6 @@ setup(
         'websocket.tests': ['data/*.txt'],
         'websocket': ["cacert.pem"]
     },
-    test_suite = "websocket.tests.test_websocket",
+    tests_require=tests_require,
+    test_suite="websocket.tests.test_websocket",
 )