Fix build

Change-Id: I55f40ae1667b0cbf5927deb8cebd9a52418f630f