diff --git a/src/Umbraco.Web.UI.Client/src/packages/core/section/section-sidebar/section-sidebar.context.ts b/src/Umbraco.Web.UI.Client/src/packages/core/section/section-sidebar/section-sidebar.context.ts
index 69d9348efb..5fad1202bb 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/core/section/section-sidebar/section-sidebar.context.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/core/section/section-sidebar/section-sidebar.context.ts
@@ -17,19 +17,19 @@ export class UmbSectionSidebarContext extends UmbContextBase<UmbSectionSidebarCo
 	#headline = new UmbStringState<undefined>(undefined);
 	headline = this.#headline.asObservable();
 
-	#contextElement: HTMLElement | undefined = undefined;
+	#contextElement: Element | undefined = undefined;
 
 	constructor(host: UmbControllerHost) {
 		super(host, UMB_SECTION_SIDEBAR_CONTEXT);
 	}
 
-	toggleContextMenu(host: HTMLElement, args: UmbOpenContextMenuArgs) {
+	toggleContextMenu(host: Element, args: UmbOpenContextMenuArgs) {
 		this.openContextMenu(host, args);
 	}
 
 	// TODO: we wont get notified about tree item name changes because we don't have a subscription
 	// we need to figure out how we best can handle this when we only know the entity and unique id
-	openContextMenu(host: HTMLElement, args: UmbOpenContextMenuArgs) {
+	openContextMenu(host: Element, args: UmbOpenContextMenuArgs) {
 		this.#entityType.setValue(args.entityType);
 		this.#unique.setValue(args.unique);
 		this.#headline.setValue(args.headline);
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 93825c96df..d86bd67532 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
@@ -180,7 +180,7 @@ export abstract class UmbTreeItemContextBase<TreeItemType extends UmbTreeItemMod
 			throw new Error('Could not request children, tree item is not set');
 		}
 
-		this.#sectionSidebarContext?.toggleContextMenu(this, {
+		this.#sectionSidebarContext?.toggleContextMenu(this.getHostElement(), {
 			entityType: this.entityType,
 			unique: this.unique,
 			headline: this.getTreeItem()?.name || '',