Revert to old (broken) version of fit.py