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

fix build err