Changes as requested in #1 comment

Two minor fixes.
Thanks, but it's such a minor addition, that it would even feel wrong ;)