Fixes #15 and now I feel like a cheat for such a simple fix :P