WPML language switcher doesn't work, repositioning of flag

Hi Ismael,

I solved the issue using your code together with one provided by WPML, see the code below.

Thanks for your help!

All the best,
Steffen

<?php

/*
* Add your own functions here. You can also copy some of the theme functions into this file. 
* WordPress will use those functions instead of the original functions then.
*/

add_action('after_setup_theme','avia_remove_main_menu_flags');
function avia_remove_main_menu_flags(){
        remove_filter( 'wp_nav_menu_items', 'avia_append_lang_flags', 20, 2 );
        remove_filter( 'avf_fallback_menu_items', 'avia_append_lang_flags', 20, 2 );
        remove_action( 'avia_meta_header', 'avia_wpml_language_switch', 10);
}

// Filter wp_nav_menu() to add additional links and other output
// Show only other language in language switcher
add_filter('wp_nav_menu_items', 'new_nav_menu_items', 10, 2);
function new_nav_menu_items($items, $args) {
    // uncomment this to find your theme's menu location
    //echo "args: <pre>"; print_r($args); echo "</pre>";
     
    // add $args->theme_location == 'primary-menu' in the conditional if we want to specify the menu location.
    // if (function_exists('icl_get_languages') && $args->theme_location == 'avia') {
    if (function_exists('icl_get_languages')) {
        $languages = icl_get_languages('skip_missing=0');
 
        if(!empty($languages)){
 
            foreach($languages as $l){
                if(!$l['active']){
                    // flag with native name
                    //$items = $items . '<li class="menu-item"><a href="' . $l['url'] . '"><img src="' . $l['country_flag_url'] . '" height="12" alt="' . $l['language_code'] . '" width="18" /> ' . $l['native_name'] . '</a></li>';
                    // only flag
                    $items = $items . '<li class="menu-item menu-item-language"><a href="' . $l['url'] . '"><img src="' . $l['country_flag_url'] . '" height="12" alt="' . $l['language_code'] . '" width="18" /></a></li>';
                }
            }
        }
    }
     
    return $items;
}