Fixes #13

Good catch @fetz