fix #9
Merge pull request #13 from PuZheng/issue9

fix #9