fix #48
Merge pull request #50 from PuZheng/issue48

fix #48