Tried to fix setup.py