Fixing the setup.py