Update setup.py

I forgot to do this earlier. Sorry JMMB!