Modify the fitting pipeline