diff --git a/src/Billing/Services/Implementations/SetupIntentSucceededHandler.cs b/src/Billing/Services/Implementations/SetupIntentSucceededHandler.cs index 78a84bc3a2..bc3fa1bd56 100644 --- a/src/Billing/Services/Implementations/SetupIntentSucceededHandler.cs +++ b/src/Billing/Services/Implementations/SetupIntentSucceededHandler.cs @@ -42,14 +42,8 @@ public class SetupIntentSucceededHandler( var organization = await organizationRepository.GetByIdAsync(subscriberId.Value); var provider = await providerRepository.GetByIdAsync(subscriberId.Value); - if (organization != null) - { - await SetPaymentMethodAsync(organization, setupIntent.PaymentMethod); - } - else if (provider != null) - { - await SetPaymentMethodAsync(provider, setupIntent.PaymentMethod); - } + OneOf entity = organization != null ? organization : provider!; + await SetPaymentMethodAsync(entity, setupIntent.PaymentMethod); } private async Task SetPaymentMethodAsync(