fix #171
Merge pull request #184 from xiechao06/issue171

fix #171