small change to model validation