diff --git a/katsu/modules/base/base-disk-aarch64.yaml b/katsu/modules/base/base-disk-aarch64.yaml
index 346777f..abf8552 100644
--- a/katsu/modules/base/base-disk-aarch64.yaml
+++ b/katsu/modules/base/base-disk-aarch64.yaml
@@ -23,16 +23,19 @@ disk:
   size: 8GiB
   partitions:
     - label: EFI
+      type: esp
       size: 512MiB
       filesystem: efi
       mountpoint: /boot/efi
 
     - label: boot
+      type: xbootldr
       size: 1GiB
       filesystem: ext4
       mountpoint: /boot
 
     - label: root
+      type: root
       # size: 2.5MiB
       filesystem: ext4
       mountpoint: /
diff --git a/katsu/modules/base/base-disk-x86_64.yaml b/katsu/modules/base/base-disk-x86_64.yaml
index 823c87e..0113d48 100644
--- a/katsu/modules/base/base-disk-x86_64.yaml
+++ b/katsu/modules/base/base-disk-x86_64.yaml
@@ -23,16 +23,19 @@ disk:
   size: 8GiB
   partitions:
     - label: EFI
+      type: esp
       size: 512MiB
       filesystem: efi
       mountpoint: /boot/efi
 
     - label: boot
+      type: xbootldr
       size: 1GiB
       filesystem: ext4
       mountpoint: /boot
 
     - label: root
+      type: root
       # size: 2.5MiB
       filesystem: ext4
       mountpoint: /