From 37c9cb10ed712b63d3606df4d55de76b858a8305 Mon Sep 17 00:00:00 2001
From: Kroese <kroese@users.noreply.github.com>
Date: Mon, 3 Aug 2026 03:33:25 +0200
Subject: [PATCH] Update answer.sh

---
 src/answer.sh | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/src/answer.sh b/src/answer.sh
index da75ba68..adb51d4f 100644
--- a/src/answer.sh
+++ b/src/answer.sh
@@ -1387,12 +1387,17 @@ updateDiskID() {
   current="$ids"
   [ "$current" = "$target" ] && return 0
 
-  if [ "$current" != "1" ]; then
-    error "The answer file must use DiskID 1 as its template value: $asset"
-    return 1
-  fi
+  case "$current" in
+    "0" | "1" ) ;;
+    * )
+      error "Unsupported DiskID $current in answer file: $asset"
+      return 1
+      ;;
+  esac
 
-  if ! sed -i 's#<DiskID>1</DiskID>#<DiskID>0</DiskID>#g' "$asset"; then
+  if ! sed -i -E \
+    "s#<DiskID>[[:space:]]*$current[[:space:]]*</DiskID>#<DiskID>$target</DiskID>#g" \
+    "$asset"; then
     error "Failed to update DiskID in answer file: $asset"
     return 1
   fi