diff --git a/ubuntu24.04/nvidia-driver b/ubuntu24.04/nvidia-driver
index 18345f751..5b47129bf 100755
--- a/ubuntu24.04/nvidia-driver
+++ b/ubuntu24.04/nvidia-driver
@@ -813,14 +813,14 @@ reload_nvidia_peermem() {
     # If nvidia-peermem is already loaded (e.g. fast-path restart), skip modprobe
     if [ -f /sys/module/nvidia_peermem/refcnt ]; then
         echo "nvidia-peermem module already loaded, skipping modprobe"
-        sleep inf
         trap "echo 'Caught signal'; exit 1" HUP INT QUIT PIPE TERM
+        sleep infinity
     fi
     if chroot /run/nvidia/driver modprobe nvidia-peermem "${NVIDIA_PEERMEM_MODULE_PARAMS[@]}"; then
         if [ -f /sys/module/nvidia_peermem/refcnt ]; then
             echo "successfully loaded nvidia-peermem module, now waiting for signal"
-            sleep inf
             trap "echo 'Caught signal'; exit 1" HUP INT QUIT PIPE TERM
+            sleep infinity
         fi
     fi
     echo "failed to load nvidia-peermem module"