Fix category bug