From 0ef192630b38e8976f5114b9dea3c571451a888d Mon Sep 17 00:00:00 2001
From: andrewwhitecdw <andrewwhitecdw@users.noreply.github.com>
Date: Wed, 12 Aug 2026 19:04:40 -0500
Subject: [PATCH] fix: set -e bypasses vgpu-util count error handler

Signed-off-by: andrewwhitecdw <andrewwhitecdw@users.noreply.github.com>
---
 ubuntu22.04/nvidia-driver | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/ubuntu22.04/nvidia-driver b/ubuntu22.04/nvidia-driver
index c1acbb027..0d58a882d 100755
--- a/ubuntu22.04/nvidia-driver
+++ b/ubuntu22.04/nvidia-driver
@@ -622,8 +622,7 @@ _find_vgpu_driver_version() {
         return 0
     fi
     # check if vgpu devices are present
-    count=$(vgpu-util count)
-    if [ $? -ne 0 ]; then
+    if ! count=$(vgpu-util count); then
          echo "cannot find vgpu devices on host, pleae check /var/log/vgpu-util.log for more details..."
          return 0
     fi