diff --git a/src/Umbraco.Web.UI.Client/src/backoffice/trees/shared/tree-item.element.ts b/src/Umbraco.Web.UI.Client/src/backoffice/trees/shared/tree-item.element.ts
index 5ea8098662..bca705c8cc 100644
--- a/src/Umbraco.Web.UI.Client/src/backoffice/trees/shared/tree-item.element.ts
+++ b/src/Umbraco.Web.UI.Client/src/backoffice/trees/shared/tree-item.element.ts
@@ -11,7 +11,7 @@ import { UmbTreeContextMenuService } from './context-menu/tree-context-menu.serv
 import type { Entity, ManifestSection } from '@umbraco-cms/models';
 import { UmbObserverMixin } from '@umbraco-cms/observable-api';
 import { UmbContextConsumerMixin } from '@umbraco-cms/context-api';
-import { UmbDataStore } from 'src/core/stores/store';
+import { UmbTreeDataStore } from 'src/core/stores/store';
 
 @customElement('umb-tree-item')
 export class UmbTreeItem extends UmbContextConsumerMixin(UmbObserverMixin(LitElement)) {
@@ -39,7 +39,7 @@ export class UmbTreeItem extends UmbContextConsumerMixin(UmbObserverMixin(LitEle
 	private _isActive = false;
 
 	private _treeContext?: UmbTreeContextBase;
-	private _store?: UmbDataStore<unknown>;
+	private _store?: UmbTreeDataStore<unknown>;
 	private _sectionContext?: UmbSectionContext;
 	private _treeContextMenuService?: UmbTreeContextMenuService;
 
@@ -52,7 +52,7 @@ export class UmbTreeItem extends UmbContextConsumerMixin(UmbObserverMixin(LitEle
 			this._observeIsSelected();
 		});
 
-		this.consumeContext('umbStore', (store: UmbDataStore<unknown>) => {
+		this.consumeContext('umbStore', (store: UmbTreeDataStore<unknown>) => {
 			this._store = store;
 		});
 
diff --git a/src/Umbraco.Web.UI.Client/src/backoffice/trees/shared/tree-navigator.element.ts b/src/Umbraco.Web.UI.Client/src/backoffice/trees/shared/tree-navigator.element.ts
index c86ca40fab..fb5627dd16 100644
--- a/src/Umbraco.Web.UI.Client/src/backoffice/trees/shared/tree-navigator.element.ts
+++ b/src/Umbraco.Web.UI.Client/src/backoffice/trees/shared/tree-navigator.element.ts
@@ -8,7 +8,7 @@ import { UmbTreeContext } from '../tree.context';
 import { UmbObserverMixin } from '@umbraco-cms/observable-api';
 import { UmbContextConsumerMixin, UmbContextProviderMixin } from '@umbraco-cms/context-api';
 import type { Entity, ManifestSection, ManifestTree } from '@umbraco-cms/models';
-import { UmbDataStore } from 'src/core/stores/store';
+import { UmbTreeDataStore } from 'src/core/stores/store';
 
 import './tree-item.element';
 
@@ -39,7 +39,7 @@ export class UmbTreeNavigator extends UmbContextConsumerMixin(UmbContextProvider
 	@state()
 	private _href?: string;
 
-	private _store?: UmbDataStore<unknown>;
+	private _store?: UmbTreeDataStore<unknown>;
 	private _sectionContext?: UmbSectionContext;
 
 	constructor() {