Fix bug.
Merge pull request #8 from ivanzhaowy/master

Fix bug.