diff --git a/docs/samples/fraud_detection_demo.ipynb b/docs/samples/fraud_detection_demo.ipynb
index c0f4fb915..390e90a8a 100644
--- a/docs/samples/fraud_detection_demo.ipynb
+++ b/docs/samples/fraud_detection_demo.ipynb
@@ -1082,6 +1082,16 @@
     "y_pred"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "y_prob = clf.predict_proba(X_test)\n",
+    "y_prob"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -1096,7 +1106,7 @@
    "outputs": [],
    "source": [
     "display = PrecisionRecallDisplay.from_predictions(\n",
-    "    y_test, y_pred, name=\"HistGradientBoostingClassifier\"\n",
+    "    y_test, y_prob[:, 1], name=\"RandomForestClassifier\"\n",
     ")\n",
     "_ = display.ax_.set_title(\"Fraud Detection Precision-Recall Curve\")"
    ]
@@ -1255,7 +1265,7 @@
    "widgets": {}
   },
   "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
+   "display_name": "feathr",
    "language": "python",
    "name": "python3"
   },
@@ -1269,11 +1279,11 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.15"
+   "version": "3.8.0 (default, Nov  6 2019, 21:49:08) \n[GCC 7.3.0]"
   },
   "vscode": {
    "interpreter": {
-    "hash": "e34a1a57d2e174682770a82d94a178aa36d3ccfaa21227c5d2308e319b7ae532"
+    "hash": "ddb0e38f168d5afaa0b8ab4851ddd8c14364f1d087c15de6ff2ee5a559aec1f2"
    }
   }
  },
diff --git a/docs/samples/product_recommendation_demo_advanced.ipynb b/docs/samples/product_recommendation_demo_advanced.ipynb
index a0fc34988..8afffffc8 100644
--- a/docs/samples/product_recommendation_demo_advanced.ipynb
+++ b/docs/samples/product_recommendation_demo_advanced.ipynb
@@ -1208,7 +1208,7 @@
             ")\n",
             "\n",
             "client.materialize_features(settings, allow_materialize_non_agg_feature=True)\n",
-            "client.wait_job_to_finish(timeout_sec=1000)"
+            "client.wait_job_to_finish(timeout_sec=5000)"
          ]
       },
       {
@@ -1329,7 +1329,7 @@
          "widgets": {}
       },
       "kernelspec": {
-         "display_name": "feathr",
+         "display_name": "Python 3",
          "language": "python",
          "name": "python3"
       },
@@ -1343,11 +1343,11 @@
          "name": "python",
          "nbconvert_exporter": "python",
          "pygments_lexer": "ipython3",
-         "version": "3.10.8"
+         "version": "3.8.5 (default, Jan 27 2021, 15:41:15) \n[GCC 9.3.0]"
       },
       "vscode": {
          "interpreter": {
-            "hash": "e34a1a57d2e174682770a82d94a178aa36d3ccfaa21227c5d2308e319b7ae532"
+            "hash": "916dbcbb3f70747c44a77c7bcd40155683ae19c65e1c03b4aa3499c5328201f1"
          }
       }
    },