Fixed the categorization