Update setup.py

Fix setup.py