WordPress - show different menus for logged in users

functions.php

// Different menus to logged in users
function md_nav_menu_args( $args = '' ) {
  if( is_user_logged_in() ) {
  // Menu for logged users
  $args['menu'] = 22;
   
  } else {
  // Menu for non-logged users
  $args['menu'] = 23;
  }
  return $args;
  }
  add_filter( 'wp_nav_menu_args', 'md_nav_menu_args' );