update code for change reference model