Fix build

Change-Id: I3d3ed2335012bb40d76650685fd0fa0a7963f218