fix #353
Merge pull request #354 from xiechao06/issue353

fix #353