diff --git a/templates/typescript/gemini-cua/index.ts b/templates/typescript/gemini-cua/index.ts
index 2d6997f..b3a915e 100644
--- a/templates/typescript/gemini-cua/index.ts
+++ b/templates/typescript/gemini-cua/index.ts
@@ -49,11 +49,9 @@ async function runStagehandTask(invocationId?: string): Promise<SearchQueryOutpu
   const stagehand = new Stagehand({
     env: "LOCAL",
     verbose: 1,
-    domSettleTimeoutMs: 30_000,
-    modelName: "gpt-4o",
-    modelClientOptions: {
-      apiKey: OPENAI_API_KEY
-    },
+    domSettleTimeout: 30_000,
+    model: "openai/gpt-4.1",
+    apiKey: OPENAI_API_KEY,
     localBrowserLaunchOptions: {
       cdpUrl: kernelBrowser.cdp_ws_url
     }
@@ -64,7 +62,7 @@ async function runStagehandTask(invocationId?: string): Promise<SearchQueryOutpu
   // Your Stagehand implementation here
   /////////////////////////////////////
   try {
-    const page = stagehand.page;
+    const page = stagehand.context.pages()[0];
 
     const agent = stagehand.agent({
       provider: "google",
diff --git a/templates/typescript/gemini-cua/package.json b/templates/typescript/gemini-cua/package.json
index 4e0c5a6..5fffa31 100644
--- a/templates/typescript/gemini-cua/package.json
+++ b/templates/typescript/gemini-cua/package.json
@@ -7,8 +7,8 @@
     "typescript": "^5"
   },
   "dependencies": {
-    "@browserbasehq/stagehand": "^2.5.2",
-    "@onkernel/sdk": "^0.15.0",
-    "zod": "^3.25.67"
+    "@browserbasehq/stagehand": "^3.0.1",
+    "@onkernel/sdk": "^0.18.0",
+    "zod": "=3.25.67"
   }
 }