fix #199
Merge pull request #201 from xiechao06/issue199

fix #199