Fixed the display for categorization