fix bug
Merge pull request #22 from georgeye/master

fix bug