diff --git a/src/install.sh b/src/install.sh
index 80159f8..003b302 100644
--- a/src/install.sh
+++ b/src/install.sh
@@ -681,9 +681,10 @@ updateXML() {
     sed -i "s/SERVERSTANDARD<\/Value>/SERVER${EDITION^^}<\/Value>/g" "$asset"
   fi
 
-  sed -i '/<ProductKey>/,/<\/ProductKey>/d' "$asset"
-  cat "$asset"
-  exit
+  if [ -n "$KEY" ]; then
+    sed -i '/<ProductKey>/,/<\/ProductKey>/d' "$asset"
+    sed -i "s/<\/UserData>/  <ProductKey>\n          <Key>${KEY}<\/Key>\n          <WillShowUI>OnError<\/WillShowUI>\n        <\/ProductKey>\n      <\/UserData>/g" "$asset"
+  fi
 
   return 0
 }