diff --git a/src/Umbraco.Web.UI.Client/libs/extensions-registry/store.models.ts b/src/Umbraco.Web.UI.Client/libs/extensions-registry/store.models.ts
index 496251134f..18e8c10c17 100644
--- a/src/Umbraco.Web.UI.Client/libs/extensions-registry/store.models.ts
+++ b/src/Umbraco.Web.UI.Client/libs/extensions-registry/store.models.ts
@@ -1,10 +1,10 @@
 import type { ManifestClass } from './models';
-import { UmbStoreBase, UmbEntityTreeStore } from '@umbraco-cms/backoffice/store';
+import { UmbStoreBase, UmbTreeStore } from '@umbraco-cms/backoffice/store';
 
 export interface ManifestStore extends ManifestClass<UmbStoreBase> {
 	type: 'store';
 }
 
-export interface ManifestTreeStore extends ManifestClass<UmbEntityTreeStore> {
+export interface ManifestTreeStore extends ManifestClass<UmbTreeStore> {
 	type: 'treeStore';
 }