Minor setup.py fix