diff --git a/docs/getting_started.ipynb b/docs/getting_started.ipynb
index 8ae6fed24..21436327e 100644
--- a/docs/getting_started.ipynb
+++ b/docs/getting_started.ipynb
@@ -1145,6 +1145,7 @@
         }
       ],
       "source": [
+        "# NBVAL_SKIP\n",
         "from pydantic import BaseModel\n",
         "\n",
         "\n",
@@ -2885,7 +2886,6 @@
         }
       ],
       "source": [
-        "# NBVAL_SKIP\n",
         "from llama_stack_client.lib.agents.agent import Agent\n",
         "from llama_stack_client.lib.agents.event_logger import EventLogger\n",
         "from llama_stack_client.types.agent_create_params import AgentConfig\n",
@@ -4326,7 +4326,7 @@
       "provenance": []
     },
     "kernelspec": {
-      "display_name": "toolchain",
+      "display_name": "master",
       "language": "python",
       "name": "python3"
     },