Fix set_menu()