fix compiling errors
Merge pull request #8 from jiezhi320/master

fix compiling errors