Change navbar links based on customer tag?

Store Design
8

{% assign menu_handle = ‘main-menu’ %}> {% if customer %}> {% if customer.tags contains ‘wholesale’ %}> {% assign menu_handle = ‘main-menu-wholesale’ %}> {% endif %}> {% endif %}

I have added this code to the header.liquid file to change the navbar according to the customer tag, but this does not seem to work.