Minor fix to setup.py