Only the ProbCut like change