Fix build

Change-Id: I2f65d0128ef0b885603caf1100a194626d04cbed