fix bug

fix bug
Merge pull request #1 from wolfand11/master

fix a bug