Исключить состояние только из варианта доставки в WooCommerce

Question

Я хочу удалить определенное состояние только из поля состояния доставки. Но он останется в состоянии выставления счетов.

Но мой текущий код удалил состояние из списка обоих состояний. Есть ли способ решить эту проблему?

Вот мой текущий код

function remove_states_for_us( $states ) {
    $not_allowed_us_states = array('MA'); 

    // Loop through not allowed us states and remove them
    foreach( $not_allowed_us_states as $state_code ) {
        if( isset($states['US'][$state_code]) ){
            unset( $states['US'][$state_code] );
        }
    }
    return $states;
}
add_filter( 'woocommerce_states', 'remove_states_for_us', 10, 1 );