fix build fail issue.

lack of including header file.