fixed the category bug