fix #173
Merge pull request #179 from xiechao06/issue178

fix #178