Fix incorrect model definition