change back to linear model