fix #2
Merge pull request #3 from xuwei-k/issue2

fix #2