Fixed the setup.py to work for me