Update setup.py

Missed out on one change.