From e01b1669c85e224235eaade007d22932703a9665 Mon Sep 17 00:00:00 2001
From: CanbiZ <47820557+MickLesk@users.noreply.github.com>
Date: Wed, 6 Aug 2025 11:08:39 +0200
Subject: [PATCH] Update post-pve-install.sh

---
 tools/pve/post-pve-install.sh | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tools/pve/post-pve-install.sh b/tools/pve/post-pve-install.sh
index ddbcdb88..b51a7a27 100644
--- a/tools/pve/post-pve-install.sh
+++ b/tools/pve/post-pve-install.sh
@@ -307,6 +307,7 @@ post_routines_common() {
     rm /etc/apt/apt.conf.d/no-nag-script 2>/dev/null
     ;;
   esac
+  echo "BEFORE WIDGET"
   apt --reinstall install proxmox-widget-toolkit &>/dev/null || msg_error "Widget toolkit reinstall failed"
   echo "AFTER WIDGET"
   if ! systemctl is-active --quiet pve-ha-lrm; then
@@ -357,8 +358,8 @@ post_routines_common() {
   case $CHOICE in
   yes)
     msg_info "Updating Proxmox VE (Patience)"
-    apt-get update &>/dev/null
-    apt-get -y dist-upgrade &>/dev/null
+    apt update &>/dev/null || msg_error "apt update failed"
+    apt -y dist-upgrade &>/dev/null || msg_error "apt dist-upgrade failed"
     msg_ok "Updated Proxmox VE"
     ;;
   no) msg_error "Selected no to Updating Proxmox VE" ;;