Fix build

Change-Id: Ic05ffefb5671b758078cfc1c0355b41da35d96f1