Forgot to add fixed setup.py