diff --git a/libs/portal-integration-angular/src/lib/services/portal-dialog-service.stories.ts b/libs/portal-integration-angular/src/lib/services/portal-dialog-service.stories.ts
index c9d88a78..aa2bc1c0 100644
--- a/libs/portal-integration-angular/src/lib/services/portal-dialog-service.stories.ts
+++ b/libs/portal-integration-angular/src/lib/services/portal-dialog-service.stories.ts
@@ -240,7 +240,7 @@ export const CustomButtonsWithAutofocus = {
   args: {
     title: 'Custom title',
     messageOrComponent: {
-      type: ComponentToDisplayCustomButtons,
+      type: ComponentToDisplayCustomButtonsComponent,
     },
     primaryKey: {
       key: 'PRIMARY_KEY',