Heat charm when configured with traefik-route can have multiple loadbalancers (for heat-api and heat-api-cfn). So remove the assertion in wait_for_traefik on number of loadbalancers. Check on number of units against number of servers should be sufficient.