Small fix
Merge pull request #17 from eggrevolver/master

Small fix