Fix logging probs