Fixed the build
Merge pull request #36 from shiyingzheng/master

Fixed the build