fix #22
Merge pull request #29 from PuZheng/issue22

fix #22