Hacky bugfix pls
Merge pull request #2 from Breilly38/master

Hacky bugfix pls