Fix the build error

Change-Id: I343b286728c482ac09c6a553f95bfe3ab35f7ab4