diff --git a/common/gpu/nvidia/pascal/default.nix b/common/gpu/nvidia/pascal/default.nix
index acd77a2b..26a7f638 100644
--- a/common/gpu/nvidia/pascal/default.nix
+++ b/common/gpu/nvidia/pascal/default.nix
@@ -1,7 +1,8 @@
-{ lib, ... }:
+{ config, lib, ... }:
 {
   imports = [ ../. ];
 
   # The open source driver does not support Pascal GPUs.
   hardware.nvidia.open = false;
+  hardware.nvidia.package = lib.mkDefault config.boot.kernelPackages.nvidiaPackages.legacy_580;
 }