Remove warning for uniform probs