diff --git a/bootloader/utils/btn.c b/bootloader/utils/btn.c
index 678b19f..8ea5161 100644
--- a/bootloader/utils/btn.c
+++ b/bootloader/utils/btn.c
@@ -81,5 +81,8 @@ u8 btn_wait_timeout(u32 time_ms, u8 mask)
 	};
 
 	// Timed out.
-	return 0;
+	if (!single_button)
+		return (res & mask);
+	else
+		return 0; // Return no button press if single button requested.
 }