Fix build

Change-Id: If5a9d51d8522c9cb7b3d93c66baf287b29c71187