Update setup.py

Fixed setup.py