Fix broken setup.py