Update setup.py

I missed that the name had changed, don't want to break tests and integration with others