diff --git a/src/Umbraco.Web.UI.Client/.github/workflows/build_test.yml b/src/Umbraco.Web.UI.Client/.github/workflows/build_test.yml
index 0ad8187a22..66dba67bc5 100644
--- a/src/Umbraco.Web.UI.Client/.github/workflows/build_test.yml
+++ b/src/Umbraco.Web.UI.Client/.github/workflows/build_test.yml
@@ -37,6 +37,7 @@ jobs:
     - run: npm ci --no-audit --no-fund --prefer-offline
     - run: npm run lint
     - run: npm run build
+    - run: npm run generate:jsonschema:dist
     - run: sudo npx playwright install-deps
     - run: npm test
     - name: Upload Code Coverage reports
diff --git a/src/Umbraco.Web.UI.Client/.github/workflows/npm-publish-github-packages.yml b/src/Umbraco.Web.UI.Client/.github/workflows/npm-publish-github-packages.yml
index aa2ea49e5b..dc3413e2d1 100644
--- a/src/Umbraco.Web.UI.Client/.github/workflows/npm-publish-github-packages.yml
+++ b/src/Umbraco.Web.UI.Client/.github/workflows/npm-publish-github-packages.yml
@@ -36,6 +36,7 @@ jobs:
           scope: '@umbraco-cms'
       - run: npm ci
       - run: npm run build:for:npm
+      - run: npm run generate:jsonschema:dist
       - run: npm run wc-analyze
       - run: npm run wc-analyze:vscode
       - name: Version and publish
diff --git a/src/Umbraco.Web.UI.Client/package.json b/src/Umbraco.Web.UI.Client/package.json
index 8c38a33769..261a764d0d 100644
--- a/src/Umbraco.Web.UI.Client/package.json
+++ b/src/Umbraco.Web.UI.Client/package.json
@@ -62,7 +62,7 @@
 	},
 	"scripts": {
 		"dev": "vite",
-		"build": "tsc --project ./src/tsconfig.json && rollup -c ./src/rollup.config.js && npm run generate:jsonschema",
+		"build": "tsc --project ./src/tsconfig.json && rollup -c ./src/rollup.config.js",
 		"build:vite": "tsc && vite build --mode staging",
 		"build:for:static": "vite build",
 		"build:for:cms": "npm run build && node ./devops/build/copy-to-cms.js",
@@ -80,7 +80,8 @@
 		"format:fix": "npm run format -- --write",
 		"generate:api": "openapi --input https://raw.githubusercontent.com/umbraco/Umbraco-CMS/v13/dev/src/Umbraco.Cms.Api.Management/OpenApi.json --output src/external/backend-api/src --postfixServices Resource --useOptions",
 		"generate:api-dev": "openapi --input http://localhost:11000/umbraco/swagger/v1/swagger.json --output src/external/backend-api/src --postfixServices Resource --useOptions",
-		"generate:jsonschema": "typescript-json-schema --required --include \"./src/packages/core/extension-registry/*.ts\" --out dist-cms/umbraco-package-schema.json tsconfig.json UmbracoPackage",
+		"generate:jsonschema": "typescript-json-schema --required --include \"./src/packages/core/extension-registry/*.ts\"",
+		"generate:jsonschema:dist": "typescript-json-schema --required --include \"./src/packages/core/extension-registry/umbraco-package.ts\" --out dist-cms/umbraco-package-schema.json tsconfig.json UmbracoPackage",
 		"storybook": "npm run wc-analyze && storybook dev -p 6006",
 		"storybook:build": "npm run wc-analyze && storybook build",
 		"build-storybook": "npm run wc-analyze && storybook build",