diff --git a/src/install.sh b/src/install.sh
index e22b99e..2f4dbd2 100644
--- a/src/install.sh
+++ b/src/install.sh
@@ -668,6 +668,13 @@ updateXML() {
     sed -z "s/<AdministratorPassword>...............<Value \/>/<AdministratorPassword>\n              <Value>$admin<\/Value>/g" -i "$asset"
   fi
 
+  if [ -n "$EDITION" ]; then
+    sed -i "s/<Value>Windows Server 2025 SERVERSTANDARD<\/Value>/<Value>Windows Server 2025 SERVER${EDITION^^}<\/Value>/g" "$asset"
+    sed -i "s/<Value>Windows Server 2022 SERVERSTANDARD<\/Value>/<Value>Windows Server 2022 SERVER${EDITION^^}<\/Value>/g" "$asset"
+    sed -i "s/<Value>Windows Server 2019 SERVERSTANDARD<\/Value>/<Value>Windows Server 2019 SERVER${EDITION^^}<\/Value>/g" "$asset"
+    sed -i "s/<Value>Windows Server 2016 SERVERSTANDARD<\/Value>/<Value>Windows Server 2016 SERVER${EDITION^^}<\/Value>/g" "$asset"
+  fi
+
   return 0
 }