Minor change to model