Fix probability label bug.