fix compiling error caused by missing include
Merge pull request #2 from MadScripts/patch-1

fix compiling error caused by missing include