diff --git a/partition.sh b/partition.sh
index 667113b..2797b9e 100755
--- a/partition.sh
+++ b/partition.sh
@@ -100,7 +100,7 @@ fi
 
 timedatectl
 
-if [[ ! "${HOST}" == "debian-live" ]]; then
+if [[ ! "$(hostname)" == "debian-live" ]]; then
 	zgenhostid \
 		-f \
 		0x00bab10c