Fix the setup.py