diff --git a/tasks/notes.py b/tasks/notes.py
index c46785a4aed744..589483706ff91a 100644
--- a/tasks/notes.py
+++ b/tasks/notes.py
@@ -62,13 +62,13 @@ def update_changelog(ctx, release_branch, target="all", upstream="origin"):
     will be generated.
     """
 
-    new_version = deduce_version(ctx, release_branch, next_version=False)
-    new_version_int = list(map(int, new_version.split(".")))
-    if len(new_version_int) != 3:
-        print(f"Error: invalid version: {new_version_int}")
-        raise Exit(code=1)
-
     with agent_context(ctx, release_branch):
+        new_version = deduce_version(ctx, release_branch, next_version=False)
+        new_version_int = list(map(int, new_version.split(".")))
+        if len(new_version_int) != 3:
+            print(f"Error: invalid version: {new_version_int}")
+            raise Exit(code=1)
+
         # Step 1 - generate the changelogs
 
         generate_agent = target in ["all", "agent"]