Fixes needed for 2.0 release

1 files changed, 4 insertions, 4 deletions

diff --git a/setup.py b/setup.py
index 3f03dfc..747d9e5 100644
--- a/setup.py
+++ b/setup.py
@@ -6,12 +6,12 @@ import sys
 from distutils.core import setup
 
 install_requires = []
-if sys.version < '3':
+if sys.version_info[0] < 3:
     install_requires += ['pies2overrides']
-elif sys.version < '3.2':
+elif sys.version_info[1] < 2:
     install_requires += ['argparse']
 
-if sys.version < '3.4':
+if sys.version_info[0] < 3 or sys.version_info[1] < 4:
     install_requires += ['enum34']
 
 setup(name='pies',
@@ -24,4 +24,4 @@ setup(name='pies',
       license="MIT",
       install_requires=install_requires,
       requires=install_requires,
-      py_modules=['pies'])
+      packages=['pies'])