diff --git a/src/Umbraco.Web.UI.Client/src/packages/core/tree/tree-item/tree-item-base/tree-item-context-base.ts b/src/Umbraco.Web.UI.Client/src/packages/core/tree/tree-item/tree-item-base/tree-item-context-base.ts
index 69d56d8374..3650bdca50 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/core/tree/tree-item/tree-item-base/tree-item-context-base.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/core/tree/tree-item/tree-item-base/tree-item-context-base.ts
@@ -210,17 +210,29 @@ export abstract class UmbTreeItemContextBase<TreeItemType extends UmbTreeItemMod
 			this.#observeHasChildren();
 		});
 
-		this.#actionEventContext = await this.getContext(UMB_ACTION_EVENT_CONTEXT);
+		this.consumeContext(UMB_ACTION_EVENT_CONTEXT, (instance) => {
+			this.#actionEventContext?.removeEventListener(
+				UmbReloadTreeItemChildrenRequestEntityActionEvent.TYPE,
+				this.#onReloadRequest as EventListener,
+			);
 
-		this.#actionEventContext.addEventListener(
-			UmbReloadTreeItemChildrenRequestEntityActionEvent.TYPE,
-			this.#onReloadRequest as EventListener,
-		);
+			this.#actionEventContext?.removeEventListener(
+				UmbRequestReloadStructureForEntityEvent.TYPE,
+				this.#onReloadStructureRequest as unknown as EventListener,
+			);
 
-		this.#actionEventContext.addEventListener(
-			UmbRequestReloadStructureForEntityEvent.TYPE,
-			this.#onReloadStructureRequest as unknown as EventListener,
-		);
+			this.#actionEventContext = instance;
+
+			this.#actionEventContext.addEventListener(
+				UmbReloadTreeItemChildrenRequestEntityActionEvent.TYPE,
+				this.#onReloadRequest as EventListener,
+			);
+
+			this.#actionEventContext.addEventListener(
+				UmbRequestReloadStructureForEntityEvent.TYPE,
+				this.#onReloadStructureRequest as unknown as EventListener,
+			);
+		});
 	}
 
 	getTreeItem() {