Asignar un menú u otro dependiendo del tipo de usuario

function lsg_menu_dinamico_autores( $args ) {
// print_r($args);
if ($args['theme_location'] == 'primary') {
if ( is_user_logged_in() ) {
$user = wp_get_current_user();
if (in_array('author', $user->roles)) {
$args['menu'] = 'menu-registrados';
}
else {
$args['menu'] = 'menu-otros';
}
}
}
return $args;
}

add_filter('wp_nav_menu_args', 'lsg_menu_dinamico_autores');