minor model fix