fix build err
Merge pull request #8 from bofeng-song/master

fix build err