Fix bug
Merge pull request #4 from JerryLocke/master

Fix bug