Preciso desativar o método de envio específico se o usuário selecionou o pagamento "Pagamento na entrega". O problema é que o código a seguir funciona apenas se eu redefinir os transitórios do WooCommerce a cada vez e atualizar. Não funciona na seleção de usuários.

add_filter( 'woocommerce_package_rates', 'alter_shipping_methods', 100 );
function alter_shipping_methods( $rates ) {

    $chosen_gateway = WC()->session->chosen_payment_method;

    // If payment is Cash on delivery remove specific shipping 
    if($chosen_gateway == 'cod') {

        foreach ( $rates as $rate_id => $rate ) {
           if ( $rate->label === 'Hrvatska pošta' ) {
              unset( $rates[ $rate_id ] );
            }
       }

    }

    return $rates;

}

Eu tenho esse código que deve ser acionado e vejo a saída no console quando clico nas opções.

jQuery(document.body).on('change', 'input[name="payment_method"]', function() {
    console.log('Payment method changed');
    jQuery('body').trigger('update_checkout');
});

Eu tentei com isso, não funciona

function action_woocommerce_checkout_update_order_review($array, $int) {
    WC()->cart->calculate_shipping();
    return;
}
add_action('woocommerce_checkout_update_order_review', 'action_woocommerce_checkout_update_order_review', 10, 2);

E também tentei uma chamada AJAX personalizada que chama uma função PHP e, dentro deste filtro, não há resultado

add_filter( 'woocommerce_package_rates', 'alter_shipping_methods', 100 );

O que devo tentar a seguir?