Fix bug
Merge pull request #131 from dreamsxin/updated

Fix bug