fix bug
Merge pull request #24 from pihizi/master

fix bug