Fix build

Change-Id: I39e169120ba9307d986e1685f8678bea5d3949c5