[Помощь] Рабочее дерево категорий, чуть дописать

[Grunuk]

Что нужно дописать чтобы при переходе на третий уровень ссылки с дерева не пропадали?
Пример:
Уровень 1
- Уровень 2
- - Уровень 3 - если нажать на ссылку на этом уровне,
после перехода в категорию, слева в дереве будет видно только:
Уровень 1
- Уровень 2

Вот код:

<div class="cpt_category_tree">

    <ul>

      {section name=i loop=$categories_tree}
        {if $categories_tree[i].level > 0 && $categories_tree[i].level < 6}
       
          {if $categories_tree[i].slug}
            {assign var=_category_url value="?categoryID=`$categories_tree[i].categoryID`&category_slug=`$categories_tree[i].slug`"|set_query_html}
          {else}
            {assign var=_category_url value="?categoryID=`$categories_tree[i].categoryID`"|set_query_html}
          {/if}
         
            <li class="row{$categories_tree[i].level-1}{if $categoryID == $categories_tree[i].categoryID} current{else}{section name=p loop=$product_category_path}{if $categories_tree[i].categoryID == $product_category_path[p].categoryID && $categories_tree[i].level == 2} basic{elseif $categories_tree[i].categoryID == $product_category_path[p].categoryID} prebasic{/if}{/section}{/if}">
         
          {if $categories_tree[i.index_next].level > $categories_tree[i].level}
                <a href='{$_category_url}'><span>{$categories_tree[i].name|escape:'html'|default:"(no name)"}</span></a>{* title/father *}
               
                <ul>
          {else}
                <a href='{$_category_url}'><span>{$categories_tree[i].name|escape:'html'|default:"(no name)"}</span></a>
                </li>
          {/if}
         
          {if $categories_tree[i.index_next].level < $categories_tree[i].level}
            {section name=j loop=$categories_tree[i].level-$categories_tree[i.index_next].level}</ul></li>{/section}
          {/if}

        {/if}
      {/section}


    </ul>
</div>