diff --git a/lenovo/legion/16ach6h/edid/default.nix b/lenovo/legion/16ach6h/edid/default.nix
index 5e7d50f..481d1f5 100644
--- a/lenovo/legion/16ach6h/edid/default.nix
+++ b/lenovo/legion/16ach6h/edid/default.nix
@@ -8,10 +8,16 @@ let
   '';
 in
 {
-  hardware.firmware = [ chip_edid ];
+  hardware.display = {
+    edid.packages = [ chip_edid ];
+
+    outputs = {
+      # For some reason, the internal display is sometimes eDP-1, and sometimes it's eDP-2
+      "eDP-1".edid = "16ach6h.bin";
+      "eDP-2".edid = "16ach6h.bin";
+    };
+  };
 
-  # For some reason, the internal display is sometimes eDP-1, and sometimes it's eDP-2
-  boot.kernelParams = [ "drm.edid_firmware=eDP-1:edid/16ach6h.bin,eDP-2:edid/16ach6h.bin" ];
 
   # This fails at the moment, https://github.com/NixOS/nixos-hardware/issues/795
   # Extra refresh rates seem to work regardless