Another fix to model