diff --git a/flake.lock b/flake.lock
index 56a80e2..11e3d17 100644
--- a/flake.lock
+++ b/flake.lock
@@ -1137,16 +1137,15 @@
         "warp-tls-simple": "warp-tls-simple"
       },
       "locked": {
-        "lastModified": 1762472097,
-        "narHash": "sha256-97TvxxBmNhgKavLV1po+oXxQdkmeN2G8GeqkUK8cUWs=",
+        "lastModified": 1762567870,
+        "narHash": "sha256-ge+k1wa+pMmmAUVwLwISp7XQrp4Ls1jllDJ9eLVBvj0=",
         "owner": "juspay",
         "repo": "vira",
-        "rev": "6d4197cabc3dbf808bef16021ea9afd978f93b6e",
+        "rev": "1654d24b72eb29820aa4f52cce38dc892814b114",
         "type": "github"
       },
       "original": {
         "owner": "juspay",
-        "ref": "autobuild-improv",
         "repo": "vira",
         "type": "github"
       }
diff --git a/flake.nix b/flake.nix
index 3a15471..a8543b3 100644
--- a/flake.nix
+++ b/flake.nix
@@ -27,7 +27,7 @@
     nix-index-database.url = "github:nix-community/nix-index-database";
     nix-index-database.inputs.nixpkgs.follows = "nixpkgs";
     try.url = "github:tobi/try";
-    vira.url = "github:juspay/vira/autobuild-improv";
+    vira.url = "github:juspay/vira";
     nix-ai-tools.url = "github:numtide/nix-ai-tools";
     nix-ai-tools.inputs.nixpkgs.follows = "nixpkgs";
     landrun-nix.url = "github:srid/landrun-nix";