Fixed category selection bug