diff --git a/docs/python/README.rst b/docs/python/README.rst
index 97b1a3b34f..bf84a2b32c 100644
--- a/docs/python/README.rst
+++ b/docs/python/README.rst
@@ -26,11 +26,11 @@ replaces *scikit-learn* to compute the predictions.
     # Train a model.
     from sklearn.datasets import load_iris
     from sklearn.model_selection import train_test_split
-    from sklearn.ensemble import RandomForest
+    from sklearn.ensemble import RandomForestClassifier
     iris = load_iris()
     X, y = iris.data, iris.target
     X_train, X_test, y_train, y_test = train_test_split(X, y)
-    clr = RandomForest()
+    clr = RandomForestClassifier()
     clr.fit(X_train, y_train)
 
     # Convert into ONNX format with onnxmltools