Fixed #13 with a patch

Fixes #13