Fixes #2
Forgot to commit