fix bug
Merge pull request #7 from pihizi/master

fix bug