From 7652329ed13599b0ca0b77b658d75d9c88192d34 Mon Sep 17 00:00:00 2001
From: Egor <egor.andrianovv@gmail.com>
Date: Wed, 4 Feb 2026 05:14:54 +0300
Subject: [PATCH] Update PromoDiscountBadge.tsx

---
 src/components/PromoDiscountBadge.tsx | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

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