diff --git a/META-INF/com/google/android/updater-script b/META-INF/com/google/android/updater-script
index 6792926..e195868 100644
--- a/META-INF/com/google/android/updater-script
+++ b/META-INF/com/google/android/updater-script
@@ -3,4 +3,4 @@
 # Dummy file; update-binary is a shell script (DO NOT CHANGE)
 #
 #
-# AK_BASE_VERSION=20220302
+# AK_BASE_VERSION=20220304
diff --git a/tools/ak3-core.sh b/tools/ak3-core.sh
index 6b42e87..ad94085 100755
--- a/tools/ak3-core.sh
+++ b/tools/ak3-core.sh
@@ -771,7 +771,7 @@ setup_ak() {
   case $block in
     boot|recovery|vendor_boot)
       case $block in
-        boot) parttype="ramdisk boot BOOT LNX android_boot bootimg KERN-A kernel KERNEL";;
+        boot) parttype="ramdisk init_boot boot BOOT LNX android_boot bootimg KERN-A kernel KERNEL";;
         recovery) parttype="ramdisk_recovery recovery RECOVERY SOS android_recovery";;
         vendor_boot) parttype="vendor_boot";;
       esac;