Fix build

Change-Id: Id0f20d19ed36f94cde00096916396dcab4c2dad8