From 36285d6a64f937dabcce873d5f94a3d249427a39 Mon Sep 17 00:00:00 2001
From: Clement Bouvet <clement.bouvet@ledger.fr>
Date: Mon, 28 Nov 2022 15:55:49 +0100
Subject: [PATCH] get addr

---
 src_nbgl/ui_get_stark_public_key.c | 17 ++++++-----------
 1 file changed, 6 insertions(+), 11 deletions(-)

diff --git a/src_nbgl/ui_get_stark_public_key.c b/src_nbgl/ui_get_stark_public_key.c
index 6d5d1fd8..66ba855b 100644
--- a/src_nbgl/ui_get_stark_public_key.c
+++ b/src_nbgl/ui_get_stark_public_key.c
@@ -15,21 +15,16 @@ static void confirmTransation(void) {
 
 static void reviewChoice(bool confirm) {
   if (confirm) {
-    confirmTransation();
-  } else {
-    reviewReject();
+    // display a status page and go back to main
+    nbgl_useCaseStatus("ADDRESS\nVERIFIED",true,confirmTransation);
+  }
+  else {
+    nbgl_useCaseStatus("Address verification\ncancelled",false,reviewReject);
   }
 }
 
 static void buildScreen(void) {
-  nbgl_useCaseChoice(
-    &C_warning64px,
-    "Verify stark key",
-    strings.tmp.tmp,
-    "Approuve",
-    "Reject",
-    reviewChoice
-  );
+  nbgl_useCaseAddressConfirmation(strings.tmp.tmp, reviewChoice);
 }
 
 void ui_display_stark_public(void) {