Fix build

Change-Id: Id269022c66f69f3460da1eb71da5869bcfcfd18f