Fix build

Change-Id: Id84ff4885b8fdca9b3987702e2ff266a073381b3