Fix bug
Merge pull request #22 from ivanzhaowy/master

Fix bug