diff --git a/python/tasks/tiktok/tasks.py b/python/tasks/tiktok/tasks.py
index 6cac43d..58cfbf7 100644
--- a/python/tasks/tiktok/tasks.py
+++ b/python/tasks/tiktok/tasks.py
@@ -469,6 +469,11 @@ def process_graph(
     return dict(
         ordered_nodes=ordered_nodes,
         done_ids=done_ids,
+        done_tasks={
+            k : celery.result.AsyncResult(task_ids[k])
+            for k in nodes
+            if task_ids.get(k) in done_ids
+        },
         task_ids=task_ids,
         error_ids=error_ids,
         pending_ids=pending_ids,