Update setup.py
Fix setup.py