Fixes #17

I guess I wasn't setting it properly or something