diff --git a/src/install.sh b/src/install.sh
index c3fa8ef..816eb7b 100644
--- a/src/install.sh
+++ b/src/install.sh
@@ -756,8 +756,8 @@ updateXML() {
   local language="$2"
   local culture region user admin pass keyboard
 
-  if [ -n "${VM_NET_IP:-}" ]; then
-    sed -i "s/ 20.20.20.1 / ${VM_NET_IP%.*}.1 /g" "$asset"
+  if [ -n "${VM_NET_GATEWAY:-}" ]; then
+    sed -i "s/ 20.20.20.1 / $VM_NET_GATEWAY /g" "$asset"
   fi
 
   [ -z "$HEIGHT" ] && HEIGHT="720"