Fix bug
Merge pull request #8 from stuntguy3000/master

Fix bug