Fix
Merge pull request #24 from illyaine/master

Fix ;)