Good catch!

I didn't test this, will have a commit up soonish.