fix #341
Merge pull request #343 from xiechao06/issue341

fix #341