Fix build

Change-Id: I045ee0db5b2e6f51879dbdc6577fc46ec8b4532b