diff --git a/src/Umbraco.Web.UI.Client/src/mocks/handlers.ts b/src/Umbraco.Web.UI.Client/src/mocks/handlers.ts
index c8e3e9402e..274c2edfd0 100644
--- a/src/Umbraco.Web.UI.Client/src/mocks/handlers.ts
+++ b/src/Umbraco.Web.UI.Client/src/mocks/handlers.ts
@@ -1,6 +1,4 @@
 import { rest } from 'msw';
-
-import { components } from '../../schemas/generated-schema';
 import {
   AllowedSectionsResponse,
   ErrorResponse,
diff --git a/src/Umbraco.Web.UI.Client/src/models/index.ts b/src/Umbraco.Web.UI.Client/src/models/index.ts
index 3556428ae3..3cf0865934 100644
--- a/src/Umbraco.Web.UI.Client/src/models/index.ts
+++ b/src/Umbraco.Web.UI.Client/src/models/index.ts
@@ -6,5 +6,8 @@ export type VersionResponse = components['schemas']['VersionResponse'];
 export type ErrorResponse = components['schemas']['ErrorResponse'];
 export type UserResponse = components['schemas']['UserResponse'];
 export type AllowedSectionsResponse = components['schemas']['AllowedSectionsResponse'];
-
 export type UmbracoInstaller = components['schemas']['UmbracoInstaller'];
+
+// Models
+export type UmbracoInstallerDatabaseModel = components['schemas']['UmbracoInstallerDatabaseModel'];
+export type UmbracoInstallerUserModel = components['schemas']['UmbracoInstallerUserModel'];