fix #50
Merge pull request #63 from xiechao06/issue50

fix #50