Fix build

Change-Id: Ib9cfb0c2be41c970b01e6eda301ec057df7ffc39