more fix
Merge pull request #5 from zhujy8833/polish

more fix