Fixed loading issue
Merge pull request #7 from DetectiveSquirrel/patch-2

Fixed loading issue