search for upe element within the dom

Automattic · Jan 11, 2025 · fea511e · fea511e
1 parent 30dd8be
commit fea511e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/client/checkout/utils/upe.js b/client/checkout/utils/upe.js
@@ -379,7 +379,9 @@ export const togglePaymentMethodForCountry = ( upeElement ) => {
 	const billingCountry =
 		billingInput?.value || window?.wcpayCustomerData?.billing_country || '';
 
-	const upeContainer = upeElement?.closest( '.wc_payment_method' );
+	const upeContainer = document.querySelector(
+		'.payment_method_woocommerce_payments_' + paymentMethodType
+	);
 	if ( supportedCountries.includes( billingCountry ) ) {
 		upeContainer.style.removeProperty( 'display' );
 	} else {