Fix build

Change-Id: Ie4da47ee8e77745281961ced55efcfa380446df0