fix #1
Merge pull request #3 from xiechao06/issue1

fix #1