Fix bug with LR/RF confidence