fix bug
Merge pull request #192 from xiechao06/dev

fix bug