Fix build

Change-Id: I04d91acade283ad2b624d1b224716792937d5864