Fix comment

I think this is what you have in mind.