Bugfix
Merge pull request #7 from benrogmans/master

Bugfix :)