slight fix to model