Fixes small issue introduced by the last commit

Closes #4
Fixes #5