Always return to main menu at the end of get_wallet_address

.vscode/settings.json

@@ -5,6 +5,8 @@
     "C_Cpp.clang_format_path": "/usr/bin/clang-format",
     "editor.formatOnSave": true,
     "ledgerDevTools.appSettings": {
-        "selectedUseCase": "release"
+        "selectedUseCase": "release",
+        "selectedDevice": "Nano S",
+        "selectedVariant": "bitcoin_testnet"
     }
 }

src/handler/get_wallet_address.c

@@ -217,5 +217,13 @@ void handler_get_wallet_address(dispatcher_context_t *dc, uint8_t protocol_versi
         }
 
         SEND_RESPONSE(dc, address, address_len, SW_OK);
+
+#ifdef HAVE_NBGL
+        // Workaround for a glitch when get_wallet_address is called right after a UX flow that has
+        // a long confirmation screen (e.g. register_wallet), as processing this command sometimes
+        // lead to the "Processing..." screen not being cleared at the end of the command.
+        // This only happens on Flex and Stax.
+        ui_menu_main();
+#endif
     }
 }