Update setup.py

There was a bug in setup.py that this fixes in order to make this pip installable.