fix #4
Merge pull request #7 from PuZheng/issue4

fix #4