diff --git a/flake.nix b/flake.nix
index b5f9f5d..28bf230 100644
--- a/flake.nix
+++ b/flake.nix
@@ -63,6 +63,7 @@
           ({...}: {nixpkgs.overlays = [overlay-unstable];})
           ({...}: {nixpkgs.config.allowUnfree = true;})
           nixos-hardware.nixosModules.msi-b550-a-pro
+          nixos-hardware.nixosModules.common-gpu-nvidia
           inputs.sops-nix.nixosModules.sops
           ./modules
           ./hosts/comfy-station/configuration.nix
diff --git a/hosts/monolith/configuration.nix b/hosts/monolith/configuration.nix
index ba24c04..819d763 100644
--- a/hosts/monolith/configuration.nix
+++ b/hosts/monolith/configuration.nix
@@ -152,7 +152,7 @@
     nvidiaSettings = true;
 
     # Optionally, you may need to select the appropriate driver version for your specific GPU.
-    package = config.boot.kernelPackages.nvidiaPackages.legacy_390;
+    package = config.boot.kernelPackages.nvidiaPackages.stable;
   };
 
   # This value determines the NixOS release from which the default
@@ -204,6 +204,9 @@
   };
 
   networking.hostName = "monolith"; # Define your hostname.
+  networking.extraHosts = ''
+    127.0.0.1 monolith
+  '';
 
   # Enable networking
   networking.networkmanager.enable = true;