Fix bug
Merge pull request #130 from dreamsxin/updated

Fix bug