diff --git a/src/Umbraco.Web.UI.Client/e2e/upgrader.spec.ts b/src/Umbraco.Web.UI.Client/e2e/upgrader.spec.ts
index df76bfebfa..8af19dab8c 100644
--- a/src/Umbraco.Web.UI.Client/e2e/upgrader.spec.ts
+++ b/src/Umbraco.Web.UI.Client/e2e/upgrader.spec.ts
@@ -1,7 +1,7 @@
 import { rest } from 'msw';
 import { expect, test } from '../test';
 import { umbracoPath } from '@umbraco-cms/utils';
-import type { ProblemDetails, StatusResponse } from '@umbraco-cms/models';
+import { ProblemDetails, RuntimeLevel, ServerStatus } from '@umbraco-cms/backend-api';
 
 test.describe('upgrader tests', () => {
 	test.beforeEach(async ({ page, worker }) => {
@@ -11,8 +11,8 @@ test.describe('upgrader tests', () => {
 				return res(
 					// Respond with a 200 status code
 					ctx.status(200),
-					ctx.json<StatusResponse>({
-						serverStatus: 'must-upgrade',
+					ctx.json<ServerStatus>({
+						serverStatus: RuntimeLevel.UPGRADE,
 					})
 				);
 			})