diff --git a/src/Umbraco.Web.UI.Client/src/packages/documents/document-types/workspace/document-type-workspace.stories.ts b/src/Umbraco.Web.UI.Client/src/packages/documents/document-types/workspace/document-type-workspace.stories.ts
index 33d093e782..a708971f00 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/documents/document-types/workspace/document-type-workspace.stories.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/documents/document-types/workspace/document-type-workspace.stories.ts
@@ -1,6 +1,6 @@
 import './document-type-workspace-editor.element.js';
 import { Meta, Story } from '@storybook/web-components';
-import { treeData } from '../../../../mocks/data/document-type.data.js';
+import { data } from '../../../../mocks/data/document-type/document-type.data.js';
 import type { UmbDocumentTypeWorkspaceElement } from './document-type-workspace.element.js';
 import { html, ifDefined } from '@umbraco-cms/backoffice/external/lit';
 
@@ -11,5 +11,5 @@ export default {
 } as Meta;
 
 export const AAAOverview: Story<UmbDocumentTypeWorkspaceElement> = () =>
-	html` <umb-document-type-workspace id="${ifDefined(treeData[0].id)}"></umb-document-type-workspace>`;
+	html` <umb-document-type-workspace id="${ifDefined(data[0].id)}"></umb-document-type-workspace>`;
 AAAOverview.storyName = 'Overview';