diff --git a/src/Umbraco.Web.UI.Client/src/packages/core/tree/tree.stories.ts b/src/Umbraco.Web.UI.Client/src/packages/core/tree/tree.stories.ts
index 946a928739..2e81b4daeb 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/core/tree/tree.stories.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/core/tree/tree.stories.ts
@@ -1,14 +1,14 @@
 import type { Meta, StoryObj } from '@storybook/web-components';
 import './default/default-tree.element.js';
-import type { UmbDefaultTreeElement } from './default/default-tree.element.js';
+import type { UmbTreeElement } from './tree.element.js';
 
-const meta: Meta<UmbDefaultTreeElement> = {
+const meta: Meta<UmbTreeElement> = {
 	title: 'Components/Tree/Tree',
 	component: 'umb-tree',
 };
 
 export default meta;
-type Story = StoryObj<UmbDefaultTreeElement>;
+type Story = StoryObj<UmbTreeElement>;
 
 // TODO: This does not display anything - need help
 export const Overview: Story = {