diff --git a/setup.sh b/setup.sh
index 53c640e..7884d25 100644
--- a/setup.sh
+++ b/setup.sh
@@ -111,7 +111,7 @@ function get_drive_id () {
         printf -- '%s' "${drive_id_single}"
         return 0
     fi
-    >2 printf -- '%s\n' 'Not exactly one '"'${1:?}'"' partition entry in /dev/disk/by-id, exiting ...'
+    >2 printf -- '%s\n' 'No '"'${1:?}'"' partition entry in /dev/disk/by-partuuid, exiting ...'
     exit 77
 }
 
@@ -151,7 +151,7 @@ function set_zpool_password () {
 }
 
 function import_pool () {
-    zpool import -d '/dev/disk/by-id' -R '/mnt' "${zpool_name}" -N -f
+    zpool import -d '/dev/disk/by-partuuid' -R '/mnt' "${zpool_name}" -N -f
     zfs load-key "${zpool_name}"
 }