diff --git a/modules/customer-invoices/src/web/proformas/list/ui/components/proforma-status-badge.tsx b/modules/customer-invoices/src/web/proformas/list/ui/components/proforma-status-badge.tsx
index 0e4910fb..83d9912d 100644
--- a/modules/customer-invoices/src/web/proformas/list/ui/components/proforma-status-badge.tsx
+++ b/modules/customer-invoices/src/web/proformas/list/ui/components/proforma-status-badge.tsx
@@ -1,9 +1,11 @@
import { Tooltip, TooltipContent, TooltipTrigger } from "@repo/shadcn-ui/components";
import { cn } from "@repo/shadcn-ui/lib/utils";
-import { ReceiptTextIcon } from "lucide-react";
import { useTranslation } from "../../../../i18n";
-import { getProformaStatusIcon } from "../../../change-status/helpers";
+import {
+ getProformaStatusBadgeClassName,
+ getProformaStatusDisplayIcon,
+} from "../../../change-status/helpers";
import type { ProformaStatus } from "../../../shared";
export type ProformaStatusBadgeProps = {
@@ -14,20 +16,7 @@ export type ProformaStatusBadgeProps = {
export const ProformaStatusBadge = ({ status, className }: ProformaStatusBadgeProps) => {
const { t } = useTranslation();
const normalizedStatus = status.toLowerCase() as ProformaStatus;
- const Icon = normalizedStatus === "issued" ? ReceiptTextIcon : getProformaStatusIcon(normalizedStatus);
-
- const stylesByStatus: Record