Fixed a bug in setup.py