diff --git a/firmware/protect.c b/firmware/protect.c
index 1f778e7728..b66e3520e5 100644
--- a/firmware/protect.c
+++ b/firmware/protect.c
@@ -77,9 +77,9 @@ bool protectButton(ButtonRequestType type, bool confirm_only)
 		}
 
 		// check for Cancel / Initialize
-		if (msg_tiny_id == MessageType_MessageType_Cancel || msg_tiny_id == MessageType_MessageType_Initialize) {
-			protectAbortedByCancel = (msg_tiny_id == MessageType_MessageType_Cancel);
-			protectAbortedByInitialize = (msg_tiny_id == MessageType_MessageType_Initialize);
+		protectAbortedByCancel = (msg_tiny_id == MessageType_MessageType_Cancel);
+		protectAbortedByInitialize = (msg_tiny_id == MessageType_MessageType_Initialize);
+		if (protectAbortedByCancel || protectAbortedByInitialize) {
 			msg_tiny_id = 0xFFFF;
 			result = false;
 			break;
@@ -128,10 +128,11 @@ const char *requestPin(PinMatrixRequestType type, const char *text)
 			usbTiny(0);
 			return pma->pin;
 		}
-		if (msg_tiny_id == MessageType_MessageType_Cancel || msg_tiny_id == MessageType_MessageType_Initialize) {
+		// check for Cancel / Initialize
+		protectAbortedByCancel = (msg_tiny_id == MessageType_MessageType_Cancel);
+		protectAbortedByInitialize = (msg_tiny_id == MessageType_MessageType_Initialize);
+		if (protectAbortedByCancel || protectAbortedByInitialize) {
 			pinmatrix_done(0);
-			protectAbortedByCancel = (msg_tiny_id == MessageType_MessageType_Cancel);
-			protectAbortedByInitialize = (msg_tiny_id == MessageType_MessageType_Initialize);
 			msg_tiny_id = 0xFFFF;
 			usbTiny(0);
 			return 0;
@@ -262,9 +263,10 @@ bool protectPassphrase(void)
 			result = true;
 			break;
 		}
-		if (msg_tiny_id == MessageType_MessageType_Cancel || msg_tiny_id == MessageType_MessageType_Initialize) {
-			protectAbortedByCancel = (msg_tiny_id == MessageType_MessageType_Cancel);
-			protectAbortedByInitialize = (msg_tiny_id == MessageType_MessageType_Initialize);
+		// check for Cancel / Initialize
+		protectAbortedByCancel = (msg_tiny_id == MessageType_MessageType_Cancel);
+		protectAbortedByInitialize = (msg_tiny_id == MessageType_MessageType_Initialize);
+		if (protectAbortedByCancel || protectAbortedByInitialize) {
 			msg_tiny_id = 0xFFFF;
 			result = false;
 			break;