Fixing setup.py