Fix build

Change-Id: If9aa6e334c6af8b3915ee7e4735123163bc3775a