diff --git a/META-INF/com/google/android/updater-script b/META-INF/com/google/android/updater-script
index 7820e3a..6fc9d85 100644
--- a/META-INF/com/google/android/updater-script
+++ b/META-INF/com/google/android/updater-script
@@ -1,4 +1,4 @@
 #FLASHAFTERUPDATEV2
-# Dummy file; update-binary is a shell script. (DO NOT CHANGE)
+# Dummy file; update-binary is a shell script (DO NOT CHANGE)
 #
-# AK_BASE_VERSION=2021.07.14
+# AK_BASE_VERSION=20210721
diff --git a/tools/ak3-core.sh b/tools/ak3-core.sh
index 2c35637..274926a 100755
--- a/tools/ak3-core.sh
+++ b/tools/ak3-core.sh
@@ -657,7 +657,7 @@ reset_ak() {
 
   current=$(dirname $home/*-files/current);
   if [ -d "$current" ]; then
-    for i in $bootimg boot-new.img; do
+    for i in $bootimg $home/boot-new.img; do
       [ -e $i ] && cp -af $i $current;
     done;
     for i in $current/*; do
@@ -672,7 +672,9 @@ reset_ak() {
   else
     rm -rf $patch $home/rdtmp;
   fi;
-  ui_print " ";
+  if [ ! "$no_block_display" ]; then
+    ui_print " ";
+  fi;
   setup_ak;
 }