Fix build

Change-Id: Iefab1a6afa991ac0e8bd91a89c683094204d8737