Include necessay header file
Merge pull request #5 from xuchunyang/trivial-fix

Include necessay header file