diff --git a/nixos/loki.nix b/nixos/loki.nix
index 9fad612..d189fa8 100644
--- a/nixos/loki.nix
+++ b/nixos/loki.nix
@@ -27,7 +27,7 @@
 
   services.xserver = {
     enable = true;
-    videoDrivers = [ "nvidiaLegacy390" ];
+    videoDrivers = [ "nvidia" ];
     layout = "it";
     xkbOptions = "eurosign:e;";
     libinput.enable = true;
@@ -35,6 +35,7 @@
 
 
   hardware = {
+    nvidia.package = config.boot.kernelPackages.nvidiaPackages.legacy_470;
     nvidia.prime = {
       offload.enable = false;
       sync.enable = true;