Fix Category logic bug

Signed-off-by: Hui Zheng <fengyimzy@gmail.com>