Fix bug
Merge pull request #9 from Nuke928/master

Fix bug