Woocommerce изменить HTML на странице оформления заказа?

Question

Я пытаюсь обернуть «некоторые» поля страницы оформления заказа в div и ввести несколько кнопок, потому что я реализую некоторую проверку адреса.

По сути, для начала мне нужна оболочка вокруг компании, адресной строки 1, адресной строки 2, города и почтового индекса, чтобы я мог показать / скрыть их, но, похоже, я не могу заставить что-либо работать.Вот что я попробовал -

В моем файле фильтров:

function change_woocommerce_field_markup($field, $key, $args, $value){
    if($key === 'billing_company') {
        $field = '<div class="address-wrapper">'.$field;
    }
    else if ($key === 'billing_postcode') {
        $field = $field.'</div>';
    }
    else {
        $field = $field;
    }
    return $field;
}
add_filter('woocommerce_form_field', '\App\change_woocommerce_field_markup', 10, 4);

Поэтому я добавляю и добавляю соответствующие поля в HTML, который открывает и закрывает оболочку.Однако результирующий HTML выглядит следующим образом:

<div class="address-wrapper"></div> <p class="form-row form-row-first validate-required" id="billing_first_name_field".... <p class="form-row form-row-last validate-required" id="billing_last_name_field"....

Так что он помещает оболочку в верхнюю часть контейнера!Я не понимаю, что с этим происходит?

Спасибо, Кевин