diff --git a/flake.lock b/flake.lock
index 98e6748..cabb1bd 100644
--- a/flake.lock
+++ b/flake.lock
@@ -1801,16 +1801,15 @@
         "warp-tls-simple": "warp-tls-simple_2"
       },
       "locked": {
-        "lastModified": 1764273843,
-        "narHash": "sha256-SRUHqAOht0IOomZtvDA4lSAZ2EdYB15ppqgDG0phqXY=",
+        "lastModified": 1764274397,
+        "narHash": "sha256-/5gL8eodOV2QonCI58d4mNpBozmoIH/P2yxY+24jWMU=",
         "owner": "juspay",
         "repo": "vira",
-        "rev": "5c0fdb145803a7a88742059b976f40527cbf1e8d",
+        "rev": "658fafc0b75f25c33552c085fd4b51bd9fcbe02f",
         "type": "github"
       },
       "original": {
         "owner": "juspay",
-        "ref": "bb",
         "repo": "vira",
         "type": "github"
       }
diff --git a/flake.nix b/flake.nix
index c85d045..4afbe9a 100644
--- a/flake.nix
+++ b/flake.nix
@@ -35,7 +35,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/bb";
+    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";