fix #104
Merge pull request #112 from xiechao06/issue104

fix #104