diff --git a/docs/notebooks/Llama_Stack_Agent_Workflows.ipynb b/docs/notebooks/Llama_Stack_Agent_Workflows.ipynb
index d785c7eb0..c922c735d 100644
--- a/docs/notebooks/Llama_Stack_Agent_Workflows.ipynb
+++ b/docs/notebooks/Llama_Stack_Agent_Workflows.ipynb
@@ -71,11 +71,11 @@
     "except ImportError:\n",
     "    print(\"Not in Google Colab environment\")\n",
     "\n",
-    "# client = LlamaStackAsLibraryClient(\"fireworks\", provider_data = {\"fireworks_api_key\": os.environ['FIREWORKS_API_KEY']})\n",
-    "# _ = client.initialize()\n",
+    "client = LlamaStackAsLibraryClient(\"fireworks\", provider_data = {\"fireworks_api_key\": os.environ['FIREWORKS_API_KEY']})\n",
+    "_ = client.initialize()\n",
     "\n",
     "# Uncomment to run on a hosted Llama Stack server\n",
-    "client = LlamaStackClient(base_url=\"http://localhost:8321\")\n",
+    "# client = LlamaStackClient(base_url=\"http://localhost:8321\")\n",
     "\n",
     "MODEL_ID = \"meta-llama/Llama-3.3-70B-Instruct\"\n",
     "\n",