fix #77
Merge pull request #82 from PuZheng/issue77

fix #77