diff --git a/packages/sdk/generated/domain-map.json b/packages/sdk/generated/domain-map.json
index 1e7b00a..13cb856 100644
--- a/packages/sdk/generated/domain-map.json
+++ b/packages/sdk/generated/domain-map.json
@@ -14,9 +14,7 @@
     },
     "Project": {
       "description": "A Stitch project containing screens.",
-      "constructorParams": [
-        "projectId"
-      ],
+      "constructorParams": ["projectId"],
       "fieldMapping": {
         "projectId": {
           "from": "name",
@@ -26,10 +24,7 @@
     },
     "Screen": {
       "description": "A generated UI screen. Provides access to HTML and screenshots.",
-      "constructorParams": [
-        "projectId",
-        "screenId"
-      ],
+      "constructorParams": ["projectId", "screenId"],
       "fieldMapping": {
         "projectId": {
           "from": "projectId"
@@ -103,7 +98,7 @@
         "projection": [
           {
             "prop": "outputComponents",
-            "index": 0
+            "index": 1
           },
           {
             "prop": "design"
@@ -320,4 +315,4 @@
       }
     }
   ]
-}
\ No newline at end of file
+}
diff --git a/packages/sdk/generated/src/index.ts b/packages/sdk/generated/src/index.ts
index ee67a99..58b7ba0 100644
--- a/packages/sdk/generated/src/index.ts
+++ b/packages/sdk/generated/src/index.ts
@@ -3,8 +3,8 @@
 DO NOT EDIT — changes will be overwritten.
 
 Source: tools-manifest.json (sha256:1f84b31604f9...)
-        domain-map.json     (sha256:99b823ad9306...)
-Generated: 2026-03-19T18:56:19.253Z
+        domain-map.json     (sha256:dfac245c7cab...)
+Generated: 2026-03-22T08:59:35.189Z
  */
 export { Stitch } from "./stitch.js";
 export { Project } from "./project.js";
diff --git a/packages/sdk/generated/src/project.ts b/packages/sdk/generated/src/project.ts
index 12ef723..bb8abb2 100644
--- a/packages/sdk/generated/src/project.ts
+++ b/packages/sdk/generated/src/project.ts
@@ -3,8 +3,8 @@
 DO NOT EDIT — changes will be overwritten.
 
 Source: tools-manifest.json (sha256:1f84b31604f9...)
-        domain-map.json     (sha256:99b823ad9306...)
-Generated: 2026-03-19T18:56:19.253Z
+        domain-map.json     (sha256:dfac245c7cab...)
+Generated: 2026-03-22T08:59:35.189Z
  */
 import { type StitchToolClient } from "../../src/client.js";
 import { StitchError } from "../../src/spec/errors.js";
@@ -37,7 +37,7 @@ export class Project {
     async generate(prompt: string, deviceType?: "DEVICE_TYPE_UNSPECIFIED" | "MOBILE" | "DESKTOP" | "TABLET" | "AGNOSTIC", modelId?: "MODEL_ID_UNSPECIFIED" | "GEMINI_3_PRO" | "GEMINI_3_FLASH"): Promise<Screen> {
         try {
           const raw = await this.client.callTool<any>("generate_screen_from_text", { projectId: this.projectId, prompt, deviceType, modelId });
-          return new Screen(this.client, { ...raw.outputComponents[0].design.screens[0], projectId: this.projectId });
+          return new Screen(this.client, { ...raw.outputComponents[1].design.screens[0], projectId: this.projectId });
         } catch (error) {
           throw StitchError.fromUnknown(error);
         }
diff --git a/packages/sdk/generated/src/screen.ts b/packages/sdk/generated/src/screen.ts
index 5aca015..c95c563 100644
--- a/packages/sdk/generated/src/screen.ts
+++ b/packages/sdk/generated/src/screen.ts
@@ -3,8 +3,8 @@
 DO NOT EDIT — changes will be overwritten.
 
 Source: tools-manifest.json (sha256:1f84b31604f9...)
-        domain-map.json     (sha256:99b823ad9306...)
-Generated: 2026-03-19T18:56:19.253Z
+        domain-map.json     (sha256:dfac245c7cab...)
+Generated: 2026-03-22T08:59:35.189Z
  */
 import { type StitchToolClient } from "../../src/client.js";
 import { StitchError } from "../../src/spec/errors.js";
diff --git a/packages/sdk/generated/src/stitch.ts b/packages/sdk/generated/src/stitch.ts
index 6aaf22d..25e5393 100644
--- a/packages/sdk/generated/src/stitch.ts
+++ b/packages/sdk/generated/src/stitch.ts
@@ -3,8 +3,8 @@
 DO NOT EDIT — changes will be overwritten.
 
 Source: tools-manifest.json (sha256:1f84b31604f9...)
-        domain-map.json     (sha256:99b823ad9306...)
-Generated: 2026-03-19T18:56:19.253Z
+        domain-map.json     (sha256:dfac245c7cab...)
+Generated: 2026-03-22T08:59:35.189Z
  */
 import { type StitchToolClient } from "../../src/client.js";
 import { StitchError } from "../../src/spec/errors.js";
diff --git a/packages/sdk/generated/src/tool-definitions.ts b/packages/sdk/generated/src/tool-definitions.ts
index 15d17f3..f588810 100644
--- a/packages/sdk/generated/src/tool-definitions.ts
+++ b/packages/sdk/generated/src/tool-definitions.ts
@@ -3,8 +3,8 @@
 DO NOT EDIT — changes will be overwritten.
 
 Source: tools-manifest.json (sha256:1f84b31604f9...)
-        domain-map.json     (sha256:99b823ad9306...)
-Generated: 2026-03-19T18:56:19.253Z
+        domain-map.json     (sha256:dfac245c7cab...)
+Generated: 2026-03-22T08:59:35.189Z
  */
 /** JSON Schema property descriptor for a tool parameter. */
 export interface ToolPropertySchema {
diff --git a/packages/sdk/generated/stitch-sdk.lock b/packages/sdk/generated/stitch-sdk.lock
index fcacc2a..6ea163b 100644
--- a/packages/sdk/generated/stitch-sdk.lock
+++ b/packages/sdk/generated/stitch-sdk.lock
@@ -7,15 +7,15 @@
     "serverUrl": "https://stitch.googleapis.com/mcp"
   },
   "generated": {
-    "generatedAt": "2026-03-19T18:56:19.351Z",
-    "sourceHash": "sha256:06c97f633a04942efd348aa1634356c9f5f0fd3e16d33802bffe7e8e9a650905",
+    "generatedAt": "2026-03-22T08:59:17.473Z",
+    "sourceHash": "sha256:130655935d57c879f54ec181eb89c64e3919164181e18993f555652310a9ef61",
     "manifestHash": "sha256:1f84b31604f95580325952f0c150a3e045543fad2596e9a4d8ed15c07e0cbf9b",
-    "domainMapHash": "sha256:99b823ad930620c571a9443c7b956f90b092d626be9ce2319d7a7bfe3a2e3db4",
+    "domainMapHash": "sha256:dfac245c7cab3d5f5d25fb70bfc1dff92a4e027b6299a7071310553dd5a29f4e",
     "fileCount": 5
   },
   "domainMap": {
-    "generatedAt": "2026-03-19T18:56:19.351Z",
-    "sourceHash": "sha256:99b823ad930620c571a9443c7b956f90b092d626be9ce2319d7a7bfe3a2e3db4",
+    "generatedAt": "2026-03-22T08:59:17.473Z",
+    "sourceHash": "sha256:dfac245c7cab3d5f5d25fb70bfc1dff92a4e027b6299a7071310553dd5a29f4e",
     "manifestHash": "sha256:1f84b31604f95580325952f0c150a3e045543fad2596e9a4d8ed15c07e0cbf9b",
     "classCount": 3,
     "bindingCount": 9
diff --git a/packages/sdk/test/unit/sdk.test.ts b/packages/sdk/test/unit/sdk.test.ts
index 4c50fab..0531594 100644
--- a/packages/sdk/test/unit/sdk.test.ts
+++ b/packages/sdk/test/unit/sdk.test.ts
@@ -205,6 +205,7 @@ describe("SDK Unit Tests", () => {
 
       (mockClient.callTool as Mock).mockResolvedValue({
         outputComponents: [
+          { designSystem: { name: "ds" } },
           {
             design: {
               screens: [{ id: "new-screen-1", name: "Generated", htmlCode: "<div>test</div>", projectId }],
@@ -236,6 +237,7 @@ describe("SDK Unit Tests", () => {
       // Mock with missing screens array
       (mockClient.callTool as Mock).mockResolvedValue({
         outputComponents: [
+          { designSystem: { name: "ds" } },
           {
             design: {
               // screens is missing