Fixed setup.py bug