fix the build

Change-Id: I9ac19624a611a6468979aa8771662845356ca365