A small bug fix

I think I forgot to save at some point because this change should have
been in the last commit.