Another small fix for setup.py