fix issue-4
Merge pull request #5 from wenbing/issue-4

fix issue-4