Adjust for model change