Apparently I only needed the one line after all.
Merge pull request #20 from Hasimir/master

Apparently I only needed the one line after all.