Update setup.py

Bugfix