From d2d54c1bf1257ce81cb30b1d362f87a09cb258ca Mon Sep 17 00:00:00 2001
From: CanbiZ <47820557+MickLesk@users.noreply.github.com>
Date: Thu, 22 May 2025 10:52:20 +0200
Subject: [PATCH] Update pve-privilege-converter.sh

---
 tools/pve/pve-privilege-converter.sh | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/tools/pve/pve-privilege-converter.sh b/tools/pve/pve-privilege-converter.sh
index 6c83c10..9b84b49 100644
--- a/tools/pve/pve-privilege-converter.sh
+++ b/tools/pve/pve-privilege-converter.sh
@@ -31,6 +31,7 @@ select_container() {
   echo -e "\nChoose a Container to convert:\n"
   IFS=$'\n'
   lxc_list=$(pct list | awk '{if(NR>1)print $1 " " $3}')
+  PS3="Enter number of container to convert: "
   select opt in $lxc_list; do
     if [ -n "$opt" ]; then
       CONTAINER_ID=$(echo "$opt" | awk '{print $1}')
@@ -72,12 +73,14 @@ backup_container() {
 select_target_storage() {
   echo -e "\nSelect target storage for new container:\n"
   target_storages=$(pvesm status --content images | awk '{if(NR>1)print $1}')
+  PS3="Enter number of target storage: "
+
   select opt in $target_storages; do
-    if [ -n "$opt" ]; then
+    if [[ -z "$opt" || ! "$opt" =~ ^[0-9]+$ ]]; then
+      echo "Invalid selection. Please choose a valid number."
+    else
       TARGET_STORAGE=$opt
       break
-    else
-      echo "Invalid selection. Try again."
     fi
   done
 }