fix to accommodate change to model