fix #11
Merge pull request #12 from xiechao06/issue11

fix #11