small fix
Merge pull request #44 from xushiwei/feature/v5

small fix