fix #55
Merge pull request #66 from xiechao06/issue55

fix #55