Minor bug in setup.py