diff --git a/flake.nix b/flake.nix
index 8b6ae2e..c21312c 100644
--- a/flake.nix
+++ b/flake.nix
@@ -63,8 +63,12 @@
 
     # Private flake for sensitive configs
     private-flake = {
-      url = "github:genebean/private-flake/initial_setup";
-      inputs.nixpkgs.follows = "nixpkgs";
+      url = "github:genebean/private-flake";
+      inputs = {
+        nixpkgs.follows = "nixpkgs";
+        simple-nixos-mailserver.follows = "simple-nixos-mailserver";
+        sops-nix.follows = "sops-nix";
+      };
     };
 
     simple-nixos-mailserver = {
diff --git a/lib/mkNixosHost.nix b/lib/mkNixosHost.nix
index 128a6c5..68b050b 100644
--- a/lib/mkNixosHost.nix
+++ b/lib/mkNixosHost.nix
@@ -27,7 +27,7 @@
       }
 
       inputs.nix-flatpak.nixosModules.nix-flatpak
-
+      inputs.private-flake.nixosModules.private.ssh-keys
       inputs.sops-nix.nixosModules.sops # system wide secrets management
       ../modules/hosts/nixos # system-wide stuff
       ../modules/hosts/nixos/${hostname} # host specific stuff