fix #331
Merge pull request #333 from xiechao06/issue331

fix #331