fix #2
Merge pull request #6 from xiechao06/issue2

fix #2