Изменение порядка оформления заказа в Woocommerce и возможность настройки способов оплаты

Question

Я изменяю порядок проверки Woocommerce с помощью функций, с которыми я получил помощь здесь: Изменить раздел проверки заказа в Wooocommerce

Проблема в том, что мне также нужно отключить способ оплаты "ХПК", если страна доставки не Швеция, поэтому у меня также есть этот код на месте:

function payment_gateway_disable_by_country( $available_gateways ) {
// Abort if in admin area
if ( is_admin() ) {
    return $available_gateways;
}

$billing_country  = WC()->customer->get_country();
$shipping_country = ! empty( WC()->customer->get_shipping_country() ) ? WC()->customer->get_shipping_country() : $billing_country;



if ( isset( $available_gateways['cod'] ) && $shipping_country != 'SE' ) {
    unset( $available_gateways['cod'] );
}

return $available_gateways;
 }
  add_filter( 'woocommerce_available_payment_gateways',      'payment_gateway_disable_by_country' );

но два кода не работают вместе. Если я использую только второй код (часть способа оплаты), а не переупорядочение, он работает нормально, и способы оплаты обновляются автоматически в зависимости от выбранной страны доставки. Но если я также добавлю код переупорядочения, способы оплаты не изменятся автоматически, только при обновлении страницы. Как я могу адаптировать код переупорядочения, чтобы оба они работали вместе?