WPML Language codes in secondary menu

Hello,

I use the code below to change the behaviour from flags to language ISO code, it works fine in the primary menu. But now I would like to change it to the secondary menu at the right top. If I enable the secondary menu in the Enfold theme options, the flags appear in the secondary menu but the code below seems to be ignored. Changing “main_menu” to “secondary_menu” does not seem to solve the issue. Please help, thanks :o)

/*
* WPML language code in navigation bar 
*/

	if(!function_exists('avia_append_lang_flags'))
	{
		//first append search item to main menu
		add_filter( 'wp_nav_menu_items', 'avia_append_lang_flags', 20, 2 );
		add_filter( 'avf_fallback_menu_items', 'avia_append_lang_flags', 20, 2 );
		
		function avia_append_lang_flags( $items, $args )
		{
		    if ((is_object($args) && $args->theme_location == 'avia'))
		    {
		        global $avia_config, $sitepress;

		        if(empty($avia_config['wpml_language_menu_position'])) $avia_config['wpml_language_menu_position'] = apply_filters('avf_wpml_language_switcher_position', 'main_menu');
		        if($avia_config['wpml_language_menu_position'] != 'main_menu') return $items;
		
		        $languages = icl_get_languages('skip_missing=0&orderby=custom');
		
		        if(is_array($languages))
		        {
		            foreach($languages as $lang)
		            {
		                $currentlang = (ICL_LANGUAGE_CODE == $lang['language_code']) ? 'avia_current_lang' : '';
		
		                if(is_home() || is_front_page()) $lang['url'] = $sitepress->language_url($lang['language_code']);
		
		                $items .= "<li class='av-language-switch-item language_".$lang['language_code']." $currentlang'><a href='".$lang['url']."'>";
				$items .= "	" .strtoupper($lang['language_code']). "";
		                $items .= "</a></li>";
		            }
		        }
		    }
		    return $items;
		}
	}

/*
* End of WPML language code in navigation bar 
*/