Ah, now I see your point. I've changed it to check in both places.