Fix build

Change-Id: Ibdb6e9c6773b374c91a77bf9d638e7c779378565