fix a bug

fix a bug
Merge pull request #51 from wenjun1055/patch-1

Thanks for the patch