Fix the broken setup.py