"after" tasks working

1 files changed, 5 insertions, 1 deletions

diff --git a/app/relay/runner.js b/app/relay/runner.js
index 1c98977..2459996 100644
--- a/app/relay/runner.js
+++ b/app/relay/runner.js
@@ -318,8 +318,12 @@ export function run_task_with_activity(task, module, activity, preempt=false, wa
     set_connected(false)
     // remove task from queue
     // queue.remove_task(task)
+    console.log('success?', task.success, activity)
     if (task.success && activity.after) {
-      return run_task_with_activity(task, module, activity.after, preempt, watch)
+      if (activity.after in modules[task.module].activities) {
+        const after_activity = modules[task.module].activities[activity.after]
+        return run_task_with_activity(task, module, after_activity, preempt, watch)
+      }
     }
     return run_next_task()
   }