fix #166
Merge pull request #168 from xiechao06/issue166

fix #166