A small model fix