Wordpress walker изменить политику вложенности меню (sub-menu на одном уровне)

Столкнулся с такой задачей - нужно выводить вложенность меню во вторую колонку, но - при стандартном выводе меню вордпресс имеем такую структуру

<ul class='menu'>
   <li>
      <a>ПУНКТ 1</a>
      <ul class='sub-menu'>
         <li>
             <a>ПОДПУНКТ</a>
             <ul class='sub-menu'>
               ...
             </ul>
         </li>
      </ul>
   </li>

   <li>
      ПУНКТ 2
   </li>
<ul>

И такая структура позволяет адекватно реализовать эту задачу только костылями на js.

Вопрос - как прописать класс walker чтобы получить примерно вот такую структуру ---

 <ul class='menu'>
   <li>
      <a>ПУНКТ 1</a>

      <ul class='sub-menu'>
           ...
      </ul>
      <ul class='sub-menu'>
           ...
      </ul>
   </li>

   <li>
      <a>ПУНКТ 2</a>
   </li>
<ul>

Тобиш место того чтобы уходить внутрь все саб-меню остаются на одном уровне