Fix setup.py
Fix setup.py