fix #6
Merge pull request #8 from xiechao06/issue6

fix #6