fix #45
Merge pull request #47 from PuZheng/issue45

fix #45