diff --git a/distributed/scheduler.py b/distributed/scheduler.py
index 1f9e09517e6..bda28bcf129 100644
--- a/distributed/scheduler.py
+++ b/distributed/scheduler.py
@@ -4260,7 +4260,7 @@ def update_graph(
         user_priority: int | dict[str, int] = 0,
         actors: bool | list[str] | None = None,
         fifo_timeout: float = 0.0,
-        code: str | None = None,
+        code: tuple[str] | None = None,
         annotations: dict | None = None,
         stimulus_id: str | None = None,
     ) -> None: