Fix build

Change-Id: I80ca7eb308fd1d981fddca3d1b96a2ddb7aab526