fix the category bug by chao lin