diff --git a/chroot_setup.sh b/chroot_setup.sh
index e5d8107..006ec4a 100644
--- a/chroot_setup.sh
+++ b/chroot_setup.sh
@@ -46,7 +46,7 @@ systemctl enable NetworkManager
 
 if [[ "${TPM2_ENABLE}" == true ]]; then
   echo ">>> Enrolling TPM2 key"
-  systemd-cryptenroll --tpm2-device=auto /dev/disk/by-partlabel/"Linux LUKS" || true
+  systemd-cryptenroll --tpm2-device=auto "${DISK}p2" || true
 fi
 
 echo ">>> Installing bootloader"