diff --git a/flake.nix b/flake.nix
index 6a30858..4072a6e 100644
--- a/flake.nix
+++ b/flake.nix
@@ -48,7 +48,7 @@
     lanzaboote.url = "github:nix-community/lanzaboote";
     lanzaboote.inputs.nixpkgs.follows = "nixpkgs";
 
-    nixified-ai.url = "github:nixified-ai/flake";
+    nixified-ai.url = "github:nixified-ai/flake/2";
 
     attic.url = "github:zhaofengli/attic";
     attic.inputs.nixpkgs.follows = "nixpkgs";