Update setup.py

Fix failing setup