Fix build

Change-Id: I3bcf91ecd6af91c768996799a8f01bd528fc82e4