fix bug
Merge pull request #8 from gauss1314/feature

fix bug