diff --git a/submodules/PremiumUI/Sources/PremiumGiftScreen.swift b/submodules/PremiumUI/Sources/PremiumGiftScreen.swift
index 979bfadb08..f217d65bb1 100644
--- a/submodules/PremiumUI/Sources/PremiumGiftScreen.swift
+++ b/submodules/PremiumUI/Sources/PremiumGiftScreen.swift
@@ -885,14 +885,14 @@ private final class PremiumGiftScreenComponent: CombinedComponent {
                         
             let purpose: AppStoreTransactionPurpose
             var quantity: Int32 = 1
-            if case .settings = self.source {
-                purpose = .giftCode(peerIds: self.peerIds, boostPeer: nil, currency: currency, amount: amount)
-                quantity = Int32(self.peerIds.count)
-            } else if let peerId = self.peerIds.first {
+            
+            if self.source == .profile || self.source == .attachMenu, let peerId = self.peerIds.first {
                 purpose = .gift(peerId: peerId, currency: currency, amount: amount)
             } else {
-                fatalError()
+                purpose = .giftCode(peerIds: self.peerIds, boostPeer: nil, currency: currency, amount: amount)
+                quantity = Int32(self.peerIds.count)
             }
+            
             let _ = (self.context.engine.payments.canPurchasePremium(purpose: purpose)
             |> deliverOnMainQueue).start(next: { [weak self] available in
                 if let strongSelf = self {