diff --git a/setup.py b/setup.py
index 422d2ecac2080ecfbf524ea78f047c678f8ebb3e..909ba1f78c4328e7374ce3243423e1ce72d35e1c 100644
--- a/setup.py
+++ b/setup.py
@@ -21,7 +21,7 @@ from setuptools import setup
 
 # normally we would import six.PY2 but can't yet assume that six
 # is installed here
-PY2 = sys.version_info < (3, 0)
+PY2 = (sys.version_info.major == 2)
 
 readme_dir = os.path.dirname(__file__)
 readme_filename = os.path.join(readme_dir, 'README.md')