Correction to setup.py