From 0ea5d25f692998baba2b488786e77fb6a2963cba Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Niels=20Lyngs=C3=B8?= <niels.lyngso@gmail.com>
Date: Mon, 6 Feb 2023 16:26:58 +0100
Subject: [PATCH] use createExtensionClass

---
 .../src/backoffice/shared/components/tree/tree.context.ts  | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

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;
+				}
+			})
 		}
 	}