Fixing the breakage to setup.py