Small improvement in referencing the model.