Fix setup.py hopefully