fix #39
Merge pull request #42 from PuZheng/issue39

fix #39