fix #31
Merge pull request #32 from PuZheng/issue31

fix #31