Fixes setup.py for submission