Fix build

Change-Id: Ia6c0081eb63dfb9976d824617f56347862c67dbc