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'] });