diff --git a/src/Umbraco.Web.UI.Client/.storybook/preview-head.html b/src/Umbraco.Web.UI.Client/.storybook/preview-head.html
index e20ae776b7..75a6ba9185 100644
--- a/src/Umbraco.Web.UI.Client/.storybook/preview-head.html
+++ b/src/Umbraco.Web.UI.Client/.storybook/preview-head.html
@@ -23,7 +23,7 @@
 		line-height: 1.3em;
 	}
 </style>
-
+<script src="https://cdn.jsdelivr.net/npm/msw/lib/iife/index.js"></script>
 <script>
 	(function () {
 		window.addEventListener('load', () => {
@@ -40,3 +40,4 @@
 		});
 	})();
 </script>
+
diff --git a/src/Umbraco.Web.UI.Client/.storybook/preview.js b/src/Umbraco.Web.UI.Client/.storybook/preview.js
index e0141af974..e45608f818 100644
--- a/src/Umbraco.Web.UI.Client/.storybook/preview.js
+++ b/src/Umbraco.Web.UI.Client/.storybook/preview.js
@@ -14,8 +14,8 @@ import { UmbDocumentStore } from '../src/backoffice/documents/documents/reposito
 import { UmbDocumentTreeStore } from '../src/backoffice/documents/documents/repository/document.tree.store.ts';
 
 import customElementManifests from '../dist/libs/custom-elements.json';
-import { UmbIconStore } from '../src/core/stores/icon/icon.store';
-import { onUnhandledRequest } from '../src/core/mocks/browser';
+import { UmbIconRegistry } from '../src/core//icon-registry/icon.registry';
+import { onUnhandledRequest } from '../src/core/mocks';
 import { handlers } from '../src/core/mocks/browser-handlers';
 import { UMB_MODAL_CONTEXT_TOKEN, UmbModalContext } from '../libs/modal';
 import { UmbLitElement } from '../src/core/lit-element';
@@ -23,17 +23,17 @@ import { UmbLitElement } from '../src/core/lit-element';
 import { umbExtensionsRegistry } from '../libs/extension-registry';
 
 import '../libs/context-api/provide/context-provider.element';
-import '../libs/controller/controller-host-initializer.element.ts';
+import '../libs/controller-api/controller-host-initializer.element.ts';
 import '../src/backoffice/core/components';
 
 import { manifests as documentManifests } from '../src/backoffice/documents';
 
 class UmbStoryBookElement extends UmbLitElement {
-	_umbIconStore = new UmbIconStore();
+	_umbIconRegistry = new UmbIconRegistry();
 
 	constructor() {
 		super();
-		this._umbIconStore.attach(this);
+		this._umbIconRegistry.attach(this);
 		this._registerExtensions(documentManifests);
 		this.provideContext(UMB_MODAL_CONTEXT_TOKEN, new UmbModalContext(this));
 	}