Fixes #4, I guess this broke with a change merged from the master branch