Fixed bug made in #479

I was a bit too quick with #479.