fix #123
Merge pull request #124 from xiechao06/issue123

fix #123