diff --git a/update.sh b/update.sh
index 4defd8e..0af1529 100644
--- a/update.sh
+++ b/update.sh
@@ -14,14 +14,14 @@ fi
 
 # get OS ditro and version
 OS=$(hostnamectl | awk '/Operating System/ {print $3}')
-VM=$(hostnamectl | awk '/Virtualization/ {print $2}')
+Chassis=$(hostnamectl | awk '/Chassis/ {print $2}')
 if [ "${OS}" == "Ubuntu" ]; then
 	OS_VER=$(hostnamectl | awk '/Operating System/ {print $4}')
 else
 	OS_VER=$(hostnamectl | awk '/Operating System/ {print $5}')
 fi
 
-if [ "${VM}" == "vmware" ]; then
+if [ "${Chassis}" == "vm" ]; then
 	VM="true"
 else
 	VM="false"