diff --git a/main.sh b/main.sh
index 36d4f31..f1c5dc1 100755
--- a/main.sh
+++ b/main.sh
@@ -1,7 +1,9 @@
 #! /bin/bash
 DRIVER=535
 
-apt show nvidia-driver-$DRIVER 2>&1 | grep -v "does not have a stable" | grep Version: | head -n1 | cut -f2 -d":" | tr -d ' ' > ./linux-nvidia-modules/pika_nvidia.txt
+echo "$(apt show kernel-pika 2>&1 | grep -v "does not have a stable" | grep Depends: | head -n1 | cut -f2 -d":" | cut -f1 -d"," | cut -f3,4 -d"-" | tr -d ' ')" > ./linux-nvidia-modules/KERNEL
+
+apt show linux-modules-nvidia-$DRIVER-$(./linux-nvidia-modules/KERNEL) 2>&1 | grep -v "does not have a stable" | grep Version: | head -n1 | cut -f2 -d":" | tr -d ' ' > ./linux-nvidia-modules/pika_nvidia.txt
 
 rm -rfv /etc/apt/preferences.d/*
 echo 'Pin: release c=external' > /etc/apt/preferences.d/0-a