fix #77
Merge pull request #78 from xiechao06/issue77

fix #77