fix #6
fix #6
Merge pull request #7 from PuZheng/issue6

fix #6