diff --git a/laptop_full_setup.sh b/laptop_full_setup.sh
index 1aad8c9..3cdb5e0 100755
--- a/laptop_full_setup.sh
+++ b/laptop_full_setup.sh
@@ -80,12 +80,10 @@ if ! $IS_UPDATE; then
     if command -v nvidia-smi &>/dev/null && \
        ! dpkg -l 2>/dev/null | grep -q nvidia-container-toolkit; then
         info "Installing NVIDIA Container Toolkit..."
-        distribution=$(. /etc/os-release && echo "$ID$VERSION_ID")
         curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey \
-          | sudo gpg --dearmor \
+          | sudo gpg --dearmor --yes \
             -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
-        curl -fsSL \
-          "https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list" \
+        curl -fsSL https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list \
           | sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' \
           | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
         sudo apt-get update -qq
diff --git a/local-ai-setup.sh b/local-ai-setup.sh
index 7c61c9f..263be2c 100755
--- a/local-ai-setup.sh
+++ b/local-ai-setup.sh
@@ -66,10 +66,9 @@ if ! $IS_UPDATE; then
 
   if command -v nvidia-smi &>/dev/null && ! dpkg -l 2>/dev/null | grep -q nvidia-container-toolkit; then
     info "Installing NVIDIA Container Toolkit..."
-    dist=$(. /etc/os-release && echo "$ID$VERSION_ID")
     curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey \
-      | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
-    curl -fsSL "https://nvidia.github.io/libnvidia-container/$dist/libnvidia-container.list" \
+      | sudo gpg --dearmor --yes -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
+    curl -fsSL https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list \
       | sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' \
       | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
     sudo apt-get update -qq && sudo apt-get install -y nvidia-container-toolkit