Yep, I think I removed that during an earlier revision where I was handling this in another way. I've readded it.