Fix the setup.py script