fix #94
Merge pull request #95 from PuZheng/issue94

fix #94