bugfix
Merge pull request #4 from pengzhanlee/master

bugfix