Correct the setup.py