Fix setup.py

Feel free to squash the history if it's too messy for you ;)