Fix category issue