Small change to model.