Category Listings at Top

Hi, guys. After upgrading to 2.5.1, I notice that when you go into an old draft (or go into edit an existing post), the categories that you previously selected are no longer listed at the very top the way they used to be in old versions of WP. Is there anyway to make it do this again? I have a lot of categories, and this way I don’t have to scroll through a few hundred categories to either unchecked or check or find out which category a particular post has been put into.

Thanks.