From 2dab9e476b2a95a57d4783b3135bd9a763fdf9de Mon Sep 17 00:00:00 2001
From: LukeZGD <26163116+LukeZGD@users.noreply.github.com>
Date: Fri, 1 Sep 2023 22:47:54 +0800
Subject: [PATCH] Update restore.sh

---
 restore.sh | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/restore.sh b/restore.sh
index f5b801d..533933f 100755
--- a/restore.sh
+++ b/restore.sh
@@ -2732,6 +2732,10 @@ restore_prepare_1033() {
 
 device_buttons() {
     local opt
+    if [[ $device_mode != "Normal" ]]; then
+        device_enter_mode pwnDFU
+        return
+    fi
     print "* This device needs to be in pwnDFU/kDFU mode before proceeding."
     print "* Select Y for pwnDFU mode, N for kDFU mode. Select Y if unsure."
     read -p "$(input 'Are both your home and power buttons working properly? (Y/n): ')" opt