From 1bbbc1d1fb6e68f8f573ac43d45f346430b08e63 Mon Sep 17 00:00:00 2001
From: Ruslan Zabarov <32499706+ruszabarov@users.noreply.github.com>
Date: Fri, 29 Aug 2025 20:17:43 -0400
Subject: [PATCH 1/2] fix: use type equality and enums

---
 ui/registry/checkout-dialog/checkout-dialog.tsx | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/ui/registry/checkout-dialog/checkout-dialog.tsx b/ui/registry/checkout-dialog/checkout-dialog.tsx
index 19ad077..b6326e2 100644
--- a/ui/registry/checkout-dialog/checkout-dialog.tsx
+++ b/ui/registry/checkout-dialog/checkout-dialog.tsx
@@ -12,7 +12,7 @@ import {
 import { getCheckoutContent } from "@/registry/checkout-dialog/lib/checkout-content";
 import { useCustomer } from "autumn-js/react";
 import { ArrowRight, ChevronDown, Loader2 } from "lucide-react";
-import type { CheckoutResult, ProductItem } from "autumn-js";
+import { UsageModel, type CheckoutResult, type ProductItem } from 'autumn-js';
 import {
   Accordion,
   AccordionContent,
@@ -153,7 +153,7 @@ function DueAmounts({ checkoutResult }: { checkoutResult: CheckoutResult }) {
     : undefined;
 
   const hasUsagePrice = product.items.some(
-    (item) => item.usage_model === "pay_per_use"
+    (item) => item.usage_model === UsageModel.PayPerUse
   );
 
   const showNextCycle = next_cycle && next_cycle.total !== checkoutResult.total;
@@ -209,7 +209,7 @@ function ProductItems({
       {checkoutResult?.product.items
         .filter((item) => item.type !== "feature")
         .map((item, index) => {
-          if (item.usage_model == "prepaid") {
+          if (item.usage_model === UsageModel.Prepaid) {
             return (
               <PrepaidItem
                 key={index}

From f28b4ce4f164fd2029aeb6c5a0a3475b97b07924 Mon Sep 17 00:00:00 2001
From: Ruslan Zabarov <32499706+ruszabarov@users.noreply.github.com>
Date: Fri, 29 Aug 2025 20:25:17 -0400
Subject: [PATCH 2/2] remove redundant assertion

---
 ui/registry/checkout-dialog/checkout-dialog.tsx | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ui/registry/checkout-dialog/checkout-dialog.tsx b/ui/registry/checkout-dialog/checkout-dialog.tsx
index b6326e2..0e53f41 100644
--- a/ui/registry/checkout-dialog/checkout-dialog.tsx
+++ b/ui/registry/checkout-dialog/checkout-dialog.tsx
@@ -214,7 +214,7 @@ function ProductItems({
               <PrepaidItem
                 key={index}
                 item={item}
-                checkoutResult={checkoutResult!}
+                checkoutResult={checkoutResult}
                 setCheckoutResult={setCheckoutResult}
               />
             );
@@ -346,7 +346,7 @@ const PrepaidItem = ({
         console.error(error);
         return;
       }
-      setCheckoutResult(data!);
+      setCheckoutResult(data);
     } catch (error) {
       console.error(error);
     } finally {