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

fix #2