fix #54
Merge pull request #67 from xiechao06/issue54

fix #54