diff --git a/src/Umbraco.Web.UI.Client/src/mocks/e2e-handlers.ts b/src/Umbraco.Web.UI.Client/src/mocks/e2e-handlers.ts
index 1f9cf1acc3..4de1be3966 100644
--- a/src/Umbraco.Web.UI.Client/src/mocks/e2e-handlers.ts
+++ b/src/Umbraco.Web.UI.Client/src/mocks/e2e-handlers.ts
@@ -7,6 +7,8 @@ import { handlers as publishedStatusHandlers } from './domains/published-status.
 import * as serverHandlers from './domains/server.handlers';
 import { handlers as upgradeHandlers } from './domains/upgrade.handlers';
 import { handlers as userHandlers } from './domains/user.handlers';
+import { handlers as treeHandlers } from './domains/entity.handlers';
+import { handlers as propertyEditorHandlers } from './domains/property-editor.handlers';
 
 export const handlers = [
 	serverHandlers.serverRunningHandler,
@@ -20,4 +22,6 @@ export const handlers = [
 	...documentTypeHandlers,
 	...manifestsHandlers.default,
 	...publishedStatusHandlers,
+	...treeHandlers,
+	...propertyEditorHandlers,
 ];