diff --git a/flake.lock b/flake.lock
index 3bb1d1b..4bda21b 100644
--- a/flake.lock
+++ b/flake.lock
@@ -999,15 +999,16 @@
         ]
       },
       "locked": {
-        "lastModified": 1759257139,
-        "narHash": "sha256-tsnW843izIiIxsDuem2yKKQKBJxTrQjvtVRexLi2Pvo=",
+        "lastModified": 1759257691,
+        "narHash": "sha256-3a0yZbaHlTGrJ3cxVuC4BX3mSXSRt8ujPrSv4rfATYw=",
         "owner": "juspay",
         "repo": "vertex",
-        "rev": "ed3873e1b34cf145b84ebbb5053469581b4e73b7",
+        "rev": "671393646db7f8b963243a50cda937a4874b51f2",
         "type": "github"
       },
       "original": {
         "owner": "juspay",
+        "ref": "claude2",
         "repo": "vertex",
         "type": "github"
       }
diff --git a/flake.nix b/flake.nix
index 524df00..fe9c182 100644
--- a/flake.nix
+++ b/flake.nix
@@ -26,7 +26,7 @@
     nixos-vscode-server.url = "github:nix-community/nixos-vscode-server";
     nix-index-database.url = "github:nix-community/nix-index-database";
     nix-index-database.inputs.nixpkgs.follows = "nixpkgs";
-    vertex.url = "github:juspay/vertex";
+    vertex.url = "github:juspay/vertex/claude2";
     vertex.inputs = {
       nixpkgs.follows = "nixpkgs";
       flake-parts.follows = "flake-parts";