diff --git a/partition.sh b/partition.sh
index 3324169..667113b 100755
--- a/partition.sh
+++ b/partition.sh
@@ -40,7 +40,7 @@ else
 fi
 
 if [[ "${ID}" == 'debian' ]]; then
-	if [[ ! "${HOST}" == "debian-live" ]]; then
+	if [[ ! "$(hostname)" == "debian-live" ]]; then
 		cat << EOF | tee /etc/apt/sources.list.d/contrib.sources 1> /dev/null
 Enabled:		yes
 Types:			deb