Fix build

Change-Id: Ia16cbc54ae045006e7312d61fc444835be986ad1