diff --git a/apps/web/src/app/vault/components/premium-badge.stories.ts b/apps/web/src/app/vault/components/premium-badge.stories.ts index ffe11d73868..5433dd99813 100644 --- a/apps/web/src/app/vault/components/premium-badge.stories.ts +++ b/apps/web/src/app/vault/components/premium-badge.stories.ts @@ -1,6 +1,8 @@ import { Meta, moduleMetadata, Story } from "@storybook/angular"; +import { of } from "rxjs"; import { JslibModule } from "@bitwarden/angular/jslib.module"; +import { BillingAccountProfileStateService } from "@bitwarden/common/billing/abstractions/account/billing-account-profile-state.service"; import { I18nService } from "@bitwarden/common/platform/abstractions/i18n.service"; import { MessagingService } from "@bitwarden/common/platform/abstractions/messaging.service"; import { BadgeModule, I18nMockService } from "@bitwarden/components"; @@ -34,6 +36,12 @@ export default { return new MockMessagingService(); }, }, + { + provide: BillingAccountProfileStateService, + useValue: { + hasPremiumFromAnySource$: of(false), + }, + }, ], }), ],