1 files changed, 3 insertions, 4 deletions

diff --git a/flakes/lib/flake.nix b/flakes/lib/flake.nix
index 5b78fb6..b5bd7ee 100644
--- a/flakes/lib/flake.nix
+++ b/flakes/lib/flake.nix
@@ -2,8 +2,7 @@
   inputs.nixpkgs.url = "github:NixOS/nixpkgs";
   inputs.flake-parts.url = "github:hercules-ci/flake-parts";
   inputs.disko.url = "github:nix-community/disko";
-  # replace with zhaofengli/colmena once https://github.com/zhaofengli/colmena/pull/161 is merged
-  inputs.colmena.url = "github:immae/colmena/add-lib-get-flake";
+  inputs.colmena.url = "github:zhaofengli/colmena";
   inputs.nixos-anywhere.url = "github:numtide/nixos-anywhere";
   inputs.nixos-anywhere.inputs.disko.follows = "disko";
   inputs.nixos-anywhere.inputs.flake-parts.follows = "flake-parts";
@@ -45,11 +44,11 @@
 
           };
           flake = {
-            nixosConfigurations.${name} = (colmena.lib.fromRawFlake self).nodes.${name};
+            nixosConfigurations.${name} = (colmena.lib.makeHive self.outputs.colmena).nodes.${name};
             nixosConfigurations."${name}WithEncryption" = let
               selfWithEncryption = nixpkgs.lib.recursiveUpdate self { outputs.colmena.meta.specialArgs.cryptKeyFile = "/run/decrypt-key"; };
             in
-              (colmena.lib.fromRawFlake selfWithEncryption).nodes.${name};
+              (colmena.lib.makeHive selfWithEncryption.outputs.colmena).nodes.${name};
             colmena = {
               meta.nixpkgs = nixpkgs.legacyPackages.${system};
               meta.specialArgs = moduleArgs;