fixed issue156
Merge pull request #157 from xiechao06/issue156

fixed #156