diff --git a/src/axiomatic/client.py b/src/axiomatic/client.py
index b753b19..6a67d55 100644
--- a/src/axiomatic/client.py
+++ b/src/axiomatic/client.py
@@ -223,6 +223,7 @@ def tool_exec(self, tool: str, code: str, poll_interval: int = 3, debug: bool =
                                 return result.output
                             else:
                                 return {
+                                    "job_id": job_id,
                                     "messages": output['messages'],
                                     "objects": self._load_objects_from_base64(output['objects'])
                                 }
diff --git a/src/axiomatic/magic.py b/src/axiomatic/magic.py
index 6367f50..3a5787e 100644
--- a/src/axiomatic/magic.py
+++ b/src/axiomatic/magic.py
@@ -126,6 +126,7 @@ def ax_tool(self, tool, cell):
                                 get_ipython().user_ns["tool_result"] = output
                             else:
                                 get_ipython().user_ns["tool_result"] = {
+                                    "job_id": job_id,
                                     "messages": output['messages'],
                                     "objects": self._load_objects_from_base64(output['objects'])
                                 }