Fix setup.py