diff --git a/src/Umbraco.Web.UI.Client/src/packages/core/notification/notification.context.test.ts b/src/Umbraco.Web.UI.Client/src/packages/core/notification/notification.context.test.ts
index 4df03c16dd..adf5fe6305 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/core/notification/notification.context.test.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/core/notification/notification.context.test.ts
@@ -1,12 +1,13 @@
 import { expect } from '@open-wc/testing';
 
 import { UmbNotificationHandler, UmbNotificationContext } from './index.js';
+import { UmbControllerHostElement } from '@umbraco-cms/backoffice/controller-api';
 
 describe('UmbNotificationContext', () => {
 	let notificationContext: UmbNotificationContext;
 
 	beforeEach(async () => {
-		notificationContext = new UmbNotificationContext();
+		notificationContext = new UmbNotificationContext(new UmbControllerHostElement());
 	});
 
 	describe('Public API', () => {
diff --git a/src/Umbraco.Web.UI.Client/src/packages/core/notification/stories/notification.stories.ts b/src/Umbraco.Web.UI.Client/src/packages/core/notification/stories/notification.stories.ts
index c8a9588ea9..2210177eda 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/core/notification/stories/notification.stories.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/core/notification/stories/notification.stories.ts
@@ -9,7 +9,9 @@ export default {
 	component: 'umb-notification-layout-default',
 	decorators: [
 		(story) =>
-			html`<umb-context-provider key="UmbNotificationContext" .value=${new UmbNotificationContext()}>
+			html`<umb-context-provider
+				key="UmbNotificationContext"
+				.create=${(host: any) => new UmbNotificationContext(host)}>
 				${story()}
 			</umb-context-provider>`,
 	],