fix #237
Merge pull request #239 from xiechao06/issue237

fix #237