fix #233
Merge pull request #240 from xiechao06/issue233

fix #233