update for change to model