diff --git a/src/Umbraco.Web.UI.Client/src/apps/installer/shared/utils.story-helpers.ts b/src/Umbraco.Web.UI.Client/src/apps/installer/shared/utils.story-helpers.ts
index cd4ff83ea0..f6cd8ef1d9 100644
--- a/src/Umbraco.Web.UI.Client/src/apps/installer/shared/utils.story-helpers.ts
+++ b/src/Umbraco.Web.UI.Client/src/apps/installer/shared/utils.story-helpers.ts
@@ -1,7 +1,7 @@
 import { UmbInstallerContext } from '../installer.context.js';
-import { html } from '@umbraco-cms/backoffice/external/lit';
+import { html, TemplateResult } from '@umbraco-cms/backoffice/external/lit';
 
-export const installerContextProvider = (story: any, installerContext = new UmbInstallerContext()) => html`
+export const installerContextProvider = (story: () => Node | string | TemplateResult, installerContext = new UmbInstallerContext()) => html`
 	<umb-context-provider
 		style="display: block;margin: 2rem 25%;padding: 1rem;border: 1px solid #ddd;"
 		key="umbInstallerContext"