diff --git a/colabs/openai/Fine_tune_OpenAI_with_Weights_and_Biases.ipynb b/colabs/openai/Fine_tune_OpenAI_with_Weights_and_Biases.ipynb
index d99d40c8..e646d5cd 100644
--- a/colabs/openai/Fine_tune_OpenAI_with_Weights_and_Biases.ipynb
+++ b/colabs/openai/Fine_tune_OpenAI_with_Weights_and_Biases.ipynb
@@ -42,17 +42,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!pip install -Uq openai tiktoken datasets tenacity"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# Remove once this PR: https://github.com/wandb/wandb/pull/6498 is merged\n",
-    "!pip install git+https://github.com/wandb/wandb.git"
+    "!pip install -Uq wandb openai tiktoken datasets tenacity"
    ]
   },
   {
@@ -534,10 +524,7 @@
     "    validation_file=openai_valid_file_info.id\n",
     ")\n",
     "\n",
-    "ft_job_id = openai_ft_job_info.id\n",
-    "\n",
-    "# Log to Weights and Biases\n",
-    "WandbLogger.sync(fine_tune_job_id=ft_job_id, project=WANDB_PROJECT)"
+    "ft_job_id = openai_ft_job_info.id"
    ]
   },
   {
@@ -545,7 +532,20 @@
    "metadata": {},
    "source": [
     "\n",
-    "> this takes around 5 minutes to train."
+    "> this takes around 5 minutes to train.\n",
+    "\n",
+    "### Start Weight & Biases Sync\n",
+    "Calling `WandbLogger.sync` will start polling OpenAI for the fine-tuning job results and log them when they are retrieved, see the [docs](https://docs.wandb.ai/guides/integrations/openai) for how to modify this behaviour"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Log to Weights and Biases\n",
+    "WandbLogger.sync(fine_tune_job_id=ft_job_id, project=WANDB_PROJECT)"
    ]
   },
   {