diff --git a/system_files/desktop/usr/share/ublue-os/just/custom.just b/system_files/desktop/usr/share/ublue-os/just/custom.just
index ca9e7c4a..c689a271 100644
--- a/system_files/desktop/usr/share/ublue-os/just/custom.just
+++ b/system_files/desktop/usr/share/ublue-os/just/custom.just
@@ -3,7 +3,7 @@ install-bazzite-arch:
   KARGS=$(rpm-ostree kargs)
   if grep 'nvidia' <<< ${KARGS}; then
     echo 'Installing Bazzite Arch (Nvidia)...'
-    yes | distrobox create -i ghcr.io/ublue-os/bazzite-arch -n bazzite-arch --nvidia
+    yes | distrobox create -i ghcr.io/ublue-os/bazzite-arch -n bazzite-arch --additional-flags "--runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=all -e NVIDIA_DRIVER_CAPABILITIES=all"
   else
     echo 'Installing Bazzite Arch...'
     yes | distrobox create -i ghcr.io/ublue-os/bazzite-arch -n bazzite-arch