Update setup.py

fix