Fix setup.py a bit