Fixed the category selection