--- a/taskcluster/taskgraph/action.py
+++ b/taskcluster/taskgraph/action.py
@@ -93,17 +93,20 @@ def backfill(project, job_id):
 def add_talos(decision_task_id, times=1):
     """
     Run the add-talos task.  This function implements `mach taskgraph add-talos`,
     and is responsible for
 
      * Adding all talos jobs to a push.
     """
     full_task_json = get_artifact(decision_task_id, "public/full-task-graph.json")
-    task_labels = [label for label in full_task_json if "talos" in label]
+    task_labels = [
+        label for label, task in full_task_json.iteritems()
+        if "talos_try_name" in task['attributes']
+    ]
     for time in xrange(times):
         add_tasks(decision_task_id, task_labels, '{}-'.format(time))
 
 
 def load_decisions(s, project, resultsets, filters):
     """
     Given a project, a list of revisions, and a dict of filters, return
     a list of taskIds from decision tasks.