Fix bug in setup.py