diff --git a/src/Umbraco.Web.UI.Client/.storybook/preview.js b/src/Umbraco.Web.UI.Client/.storybook/preview.js
index daf7be5ff0..222aba5c2a 100644
--- a/src/Umbraco.Web.UI.Client/.storybook/preview.js
+++ b/src/Umbraco.Web.UI.Client/.storybook/preview.js
@@ -6,10 +6,10 @@ import { html } from 'lit-html';
 import { initialize, mswDecorator } from 'msw-storybook-addon';
 
 import { UmbExtensionRegistry } from '../src/core/extension';
-import { UmbDataTypeStore } from '../src/core/stores/data-type.store';
+import { UmbDataTypeStore } from '../src/core/stores/data-type/data-type.store';
 import { UmbDocumentTypeStore } from '../src/core/stores/document-type.store';
 import { UmbNodeStore } from '../src/core/stores/node.store';
-import { UmbPropertyEditorStore } from '../src/core/stores/property-editor.store';
+import { UmbPropertyEditorStore } from '../src/core/stores/property-editor/property-editor.store';
 import { onUnhandledRequest } from '../src/mocks/browser';
 import { handlers } from '../src/mocks/browser-handlers';
 import { internalManifests } from '../src/temp-internal-manifests';