fix #78
Merge pull request #83 from PuZheng/issue78

fix #78