diff --git a/openflexure_microscope/task.py b/openflexure_microscope/task.py
index 5575c430..526cb91f 100644
--- a/openflexure_microscope/task.py
+++ b/openflexure_microscope/task.py
@@ -21,7 +21,12 @@ class TaskOrchestrator:
         """
         Returns a list of dictionary representations of all tasks in the session.
         """
-        return [task.state for task in self.tasks]
+        state = {}
+
+        for task in self.tasks:
+            state[task.id] = task.state
+
+        return state
 
     def task_from_id(self, task_id: str):
         """