Fix reference update

Change-Id: Ic93de796b914674e7302db2ff7748187a77d264b