Woocommerce – Escolher tipo de usuário ao registrar

Olá a todos,

Faz um tempo que venho buscando uma solução para que minha loja possa no momento do registro do usuário (checkou) oferecer duas opções de cadastro via combobox:

Customer: customer
Revendedor: revenda

Já tentei vários plugins mas sem sucesso. O que chegou mais próximo, foi o código abaixo dentro do arquivo functions.php do tema:

function wooc_extra_register_fields() {
       ?>
       <p class="form-row form-row-first">

       <select name="role">
       	<option value="customer">Varejo</option>
       	<option value="customer">Revendedor</option>
       </select>
       </p>
       <?php
}
add_action( 'woocommerce_register_form_start', 'wooc_extra_register_fields' );

Mas o código acima, só inclui o campo dentro do formulário de registro que fica em [woocommerce_my_account].

Não consegui incluir o mesmo campo dentro do formulário que fica em [woocommerce_checkout]

Alguém poderia me ajudar?

Tks