fix bug
Merge pull request #11 from Z-starts/master

fix bug