fix #172
Merge pull request #173 from xiechao06/issue172

fix #172