fix #43
Merge pull request #44 from PuZheng/issue43

fix #43