diff --git a/arch-setup.sh b/arch-setup.sh
index 5a9b150..aa82a77 100755
--- a/arch-setup.sh
+++ b/arch-setup.sh
@@ -131,15 +131,15 @@ gpu_setup() {
                 if pacman -Qs linux > /dev/null; then
                 sudo pacman -S --needed --noconfirm nvidia-open
                 fi
-                if pacman -Qs linux-zen > /dev/null; then
-                sudo pacman -S --needed --noconfirm nvidia-open
-                fi
                 if pacman -Qs linux-lts > /dev/null; then
                 sudo pacman -S --needed --noconfirm nvidia-open-lts
                 fi
                 if pacman -Qs linux-cachyos > /dev/null; then
                 sudo pacman -S --needed --noconfirm linux-cachyos-nvidia-open
                 fi
+                if pacman -Qs linux-zen > /dev/null || pacman -Qs linux-hardened > /dev/null; then
+                sudo pacman -S --needed --noconfirm nvidia-open-dkms
+                fi
 
                 if ! lspci | grep -Ei "amd|intel" > /dev/null; then
                 sudo pacman -S --needed --noconfirm libva-nvidia-driver