diff --git a/flake.nix b/flake.nix
index a655038..812c667 100644
--- a/flake.nix
+++ b/flake.nix
@@ -5,7 +5,7 @@
     nixpkgs.url = "github:radvendii/nixpkgs/nixos_rebuild_tests";
   };
 
-  outputs = { self, nixpkgs }:
+  outputs = inputs@{ self, nixpkgs }:
   let
     system = "x86_64-linux";
     pkgs = nixpkgs.legacyPackages.${system};
@@ -13,10 +13,10 @@
 
     nixosModules = {
       ## Fediversity modules
-      fediversity = {
+      fediversity = { pkgs, ... }: {
         imports = [ ./fediversity ];
 
-        services.pixelfed.package = pkgs.pixelfed.overrideAttrs (old: {
+        services.pixelfed.package = inputs.nixpkgs.legacyPackages.${pkgs.stdenv.hostPlatform.system}.pixelfed.overrideAttrs (old: {
           patches = (old.patches or [ ]) ++ [ ./fediversity/pixelfed-group-permissions.patch ];
         });
       };