Fixes #11 (forgot to commit)