fix #336
Merge pull request #339 from xiechao06/issue336

fix #336