Fix build

Change-Id: I580888bd331f96cd736f73a29e582695e678cd6c