fix #350
Merge pull request #352 from xiechao06/issue350

fix #350