diff --git a/tutorials/custom_botorch_model_in_ax.ipynb b/tutorials/custom_botorch_model_in_ax.ipynb
index 3d79c4163a..1387864666 100644
--- a/tutorials/custom_botorch_model_in_ax.ipynb
+++ b/tutorials/custom_botorch_model_in_ax.ipynb
@@ -1085,6 +1085,7 @@
     "    Runner,\n",
     "    SearchSpace,\n",
     ")\n",
+    "from ax.utils.common.result import Ok\n",
     "from botorch.test_functions import Branin\n",
     "\n",
     "\n",
@@ -1106,7 +1107,7 @@
     "                    \"sem\": float(\"nan\"),  # SEM (observation noise) - NaN indicates unknown\n",
     "                }\n",
     "            )\n",
-    "        return Data(df=pd.DataFrame.from_records(records))\n",
+    "        return Ok(value=Data(df=pd.DataFrame.from_records(records)))\n",
     "\n",
     "\n",
     "# Search space defines the parameters, their types, and acceptable values.\n",