diff --git a/docs/source/ml.rst b/docs/source/ml.rst
index 4696b279cbf..00a3fd50ba9 100644
--- a/docs/source/ml.rst
+++ b/docs/source/ml.rst
@@ -150,8 +150,8 @@ see which were likely to become ill
 
    # pandas code
    # predictions = model.predict(df)
-   # df["predictions"] = predictions
+   predictions.to_parquet("/path/to/results.parquet")
 
    # Dask code
    predictions = df.map_partitions(model.predict)
-   df["predictions"] = predictions
+   predictions.to_parquet("/path/to/results.parquet")