Fix build

Change-Id: I596bf1eb4831404cf034e7ff2dc46416e4eb05ae