fix bug
Merge pull request #136 from xiechao06/fix_bug

fix bug