Forgot to update the setup.py