Fix compile

Change-Id: I45ff9be4e11633204a04cf9691cd16b90c5d8be6