diff --git a/flake.nix b/flake.nix
index 7790e94..a693e7d 100644
--- a/flake.nix
+++ b/flake.nix
@@ -92,7 +92,7 @@
                 openldap = prev.openldap.overrideAttrs {
                   doCheck = !prev.stdenv.hostPlatform.isi686;
                 };
-                niri-scratchpad = inputs.niri-scratchpad.packages.${prev.system}.default;
+                niri-scratchpad = inputs.niri-scratchpad.packages.${prev.stdenv.hostPlatform.system}.default;
                 vimPlugins = prev.vimPlugins.extend (
                   f: p: {
                     neotest-pest = p.neotest-pest.overrideAttrs (_: {
@@ -108,7 +108,7 @@
               })
               inputs.nur.overlays.default
               (_: prev: {
-                niri = inputs.niri-branch.packages.${prev.system}.niri;
+                niri = inputs.niri-branch.packages.${prev.stdenv.hostPlatform.system}.niri;
               })
             ];
           }