diff --git a/setup.sh b/setup.sh
index b6bed68..a023530 100644
--- a/setup.sh
+++ b/setup.sh
@@ -113,6 +113,7 @@ function uefi_or_bios () {
             'neither. Cowardly exiting ...'
         exit 77
     fi
+    export part_schema="${part_schema}"
 }
 
 function get_partitions () {
@@ -349,11 +350,13 @@ function mount_system () {
 
     if [[ "${part_schema}" = 'mbr' ]]; then
         boot_drive="${zfs_parent}"
+        export boot_drive="${boot_drive}"
         boot_part="$(select_part 'boot' "${zfs_parent:?}")"
         mkdir -p '/mnt/boot/syslinux'
         mount "${boot_part}" '/mnt/boot/syslinux'
     else
         efi_drive="${zfs_parent}"
+        export efi_drive="${efi_drive}"
         efi_part="$(select_part 'boot' "${zfs_parent:?}")"
         mkdir -p '/mnt/efi'
         mount "${efi_part}" '/mnt/efi'