fix bug
Merge pull request #1 from pihizi/master

fix bug