fix #24
fix #24
Merge pull request #30 from PuZheng/issue24

fix #24