Fix a bug in setup.py