Fixed category bug issue