From 18bce185f03675ebbc1f53aa0936837f904329bb Mon Sep 17 00:00:00 2001
From: Alex Morask <144709477+amorask-bitwarden@users.noreply.github.com>
Date: Thu, 31 Jul 2025 08:19:20 -0500
Subject: [PATCH] Fix TaxService.previewTaxAmountForOrganizationTrial return
 type (#15848)

---
 .../trial-initiation/trial-billing-step.component.ts        | 3 ++-
 .../src/billing/abstractions/tax.service.abstraction.ts     | 3 +--
 libs/common/src/billing/services/tax.service.ts             | 6 +++---
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/apps/web/src/app/billing/accounts/trial-initiation/trial-billing-step.component.ts b/apps/web/src/app/billing/accounts/trial-initiation/trial-billing-step.component.ts
index fda7faeeb25..e13fac41f75 100644
--- a/apps/web/src/app/billing/accounts/trial-initiation/trial-billing-step.component.ts
+++ b/apps/web/src/app/billing/accounts/trial-initiation/trial-billing-step.component.ts
@@ -304,6 +304,7 @@ export class TrialBillingStepComponent implements OnInit, OnDestroy {
     this.fetchingTaxAmount = true;
 
     if (!this.taxInfoComponent.validate()) {
+      this.fetchingTaxAmount = false;
       return 0;
     }
 
@@ -326,7 +327,7 @@ export class TrialBillingStepComponent implements OnInit, OnDestroy {
 
     const response = await this.taxService.previewTaxAmountForOrganizationTrial(request);
     this.fetchingTaxAmount = false;
-    return response.taxAmount;
+    return response;
   };
 
   get price() {
diff --git a/libs/common/src/billing/abstractions/tax.service.abstraction.ts b/libs/common/src/billing/abstractions/tax.service.abstraction.ts
index 73dc848c95f..c94fbcba652 100644
--- a/libs/common/src/billing/abstractions/tax.service.abstraction.ts
+++ b/libs/common/src/billing/abstractions/tax.service.abstraction.ts
@@ -3,7 +3,6 @@ import { PreviewIndividualInvoiceRequest } from "../models/request/preview-indiv
 import { PreviewOrganizationInvoiceRequest } from "../models/request/preview-organization-invoice.request";
 import { PreviewTaxAmountForOrganizationTrialRequest } from "../models/request/tax";
 import { PreviewInvoiceResponse } from "../models/response/preview-invoice.response";
-import { PreviewTaxAmountResponse } from "../models/response/tax";
 
 export abstract class TaxServiceAbstraction {
   abstract getCountries(): CountryListItem[];
@@ -20,5 +19,5 @@ export abstract class TaxServiceAbstraction {
 
   abstract previewTaxAmountForOrganizationTrial: (
     request: PreviewTaxAmountForOrganizationTrialRequest,
-  ) => Promise<PreviewTaxAmountResponse>;
+  ) => Promise<number>;
 }
diff --git a/libs/common/src/billing/services/tax.service.ts b/libs/common/src/billing/services/tax.service.ts
index 2632ca7083b..27966016913 100644
--- a/libs/common/src/billing/services/tax.service.ts
+++ b/libs/common/src/billing/services/tax.service.ts
@@ -1,5 +1,4 @@
 import { PreviewTaxAmountForOrganizationTrialRequest } from "@bitwarden/common/billing/models/request/tax";
-import { PreviewTaxAmountResponse } from "@bitwarden/common/billing/models/response/tax";
 
 import { ApiService } from "../../abstractions/api.service";
 import { TaxServiceAbstraction } from "../abstractions/tax.service.abstraction";
@@ -306,13 +305,14 @@ export class TaxService implements TaxServiceAbstraction {
 
   async previewTaxAmountForOrganizationTrial(
     request: PreviewTaxAmountForOrganizationTrialRequest,
-  ): Promise<PreviewTaxAmountResponse> {
-    return await this.apiService.send(
+  ): Promise<number> {
+    const response = await this.apiService.send(
       "POST",
       "/tax/preview-amount/organization-trial",
       request,
       true,
       true,
     );
+    return response as number;
   }
 }