fix #96
Merge pull request #99 from xiechao06/issue96

fix #96