Fix setup.py bug