Small bug fix to setup.py.