diff --git a/src/Umbraco.Web.UI.Client/src/backoffice/shared/components/tree/tree.context.ts b/src/Umbraco.Web.UI.Client/src/backoffice/shared/components/tree/tree.context.ts
index 81fe32d80d..da5be15154 100644
--- a/src/Umbraco.Web.UI.Client/src/backoffice/shared/components/tree/tree.context.ts
+++ b/src/Umbraco.Web.UI.Client/src/backoffice/shared/components/tree/tree.context.ts
@@ -3,6 +3,7 @@ import { UmbTreeRepository } from '@umbraco-cms/repository';
 import type { ManifestTree } from '@umbraco-cms/models';
 import { DeepState } from '@umbraco-cms/observable-api';
 import { UmbControllerHostInterface } from '@umbraco-cms/controller';
+import { createExtensionClass } from 'libs/extensions-api/create-extension-class.function';
 
 export interface UmbTreeContext {
 	tree: ManifestTree;
@@ -30,7 +31,11 @@ export class UmbTreeContextBase implements UmbTreeContext {
 		this.tree = tree;
 
 		if (this.tree.meta.repository) {
-			this.repository = new this.tree.meta.repository(this.#host);
+			createExtensionClass<UmbTreeRepository>(this.tree.meta, [this.#host]).then((instance) {
+				if(instance) {
+					this.repository = instance;
+				}
+			})
 		}
 	}