Fix build err (#152)
Merge pull request #162 from 7histle/master

Fix build err (#152)