fix the category problem