From fa34be766aae1582fc0d9449194e7e65f22cc645 Mon Sep 17 00:00:00 2001
From: Egor <egor.andrianovv@gmail.com>
Date: Wed, 4 Feb 2026 05:11:36 +0300
Subject: [PATCH] Add files via upload

---
 src/components/PromoDiscountBadge.tsx | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/components/PromoDiscountBadge.tsx b/src/components/PromoDiscountBadge.tsx
index 16301d0..96baace 100644
--- a/src/components/PromoDiscountBadge.tsx
+++ b/src/components/PromoDiscountBadge.tsx
@@ -221,7 +221,13 @@ export default function PromoDiscountBadge() {
   });
 
   const deactivateMutation = useMutation({
-    mutationFn: promoApi.deactivateDiscount,
+    mutationFn: () => {
+      const source = activeDiscount?.source;
+      if (source && source.startsWith('promocode:')) {
+        return promoApi.deactivateDiscount();
+      }
+      return promoApi.clearActiveDiscount();
+    },
     onSuccess: () => {
       queryClient.invalidateQueries({ queryKey: ['active-discount'] });
       queryClient.invalidateQueries({ queryKey: ['promo-offers'] });