Fixed bug in setup.py