Fix to setup.py