Yeah, you're totally right.  My intention was to implement it exactly the way you suggest, and I have no idea why I didn't ;)