Fix build

Change-Id: I2fece91185a38022fedbfe2039a247decec3bf48