fix model fitting section