Fix build

Change-Id: I8650455c85e321dd5d8ed651a92f09f75cffdae0