fix up 2.0 probs with docs
fix up 2.0 probs with docs