Fix build

Change-Id: I90c295cdfb40c7fc145dc39a2ead5efefda99c3c