diff --git a/hosts/desktop/nvidia.nix b/hosts/desktop/nvidia.nix
index 2611b0b..e78b13c 100644
--- a/hosts/desktop/nvidia.nix
+++ b/hosts/desktop/nvidia.nix
@@ -1,4 +1,4 @@
-_: {
+{pkgsUnstable, ...}: {
   hardware.graphics = {
     enable = true;
   };
@@ -12,4 +12,7 @@ _: {
 
   # Avoid nvidia driver from crashing the system
   boot.kernelParams = ["pcie_aspm=off"];
+
+  # To fix https://forums.developer.nvidia.com/t/ubuntu-24-04-unable-to-change-power-state-from-d3cold-to-d0-device-inaccessible/304459
+  boot.kernelPackages = pkgsUnstable.linuxPackages_latest;
 }