diff --git a/home/linux/hyprland/wayland-apps.nix b/home/linux/hyprland/wayland-apps.nix
index c085c33b..b77b6dcf 100644
--- a/home/linux/hyprland/wayland-apps.nix
+++ b/home/linux/hyprland/wayland-apps.nix
@@ -20,7 +20,7 @@
     firefox = {
       enable = true;
       enableGnomeExtensions = false;
-      # package = pkgs.firefox-wayland; # firefox with wayland support
+      package = pkgs.firefox-wayland; # firefox with wayland support
     };
 
     vscode = {
diff --git a/hosts/idols/ai/default.nix b/hosts/idols/ai/default.nix
index 0fac08d8..980f30b7 100644
--- a/hosts/idols/ai/default.nix
+++ b/hosts/idols/ai/default.nix
@@ -77,8 +77,15 @@
   # for Nvidia GPU
   services.xserver.videoDrivers = ["nvidia"]; # will install nvidia-vaapi-driver by default
   hardware.nvidia = {
+    # Optionally, you may need to select the appropriate driver version for your specific GPU.
     package = config.boot.kernelPackages.nvidiaPackages.stable;
+
+    # Modesetting is needed for most Wayland compositors
     modesetting.enable = true;
+    # Use the open source version of the kernel module
+    # Only available on driver 515.43.04+
+    open = false;
+
     powerManagement.enable = true;
   };
   virtualisation.docker.enableNvidia = true; # for nvidia-docker