small model fix