diff --git a/taskcluster/self_taskgraph/custom_target_tasks.py b/taskcluster/self_taskgraph/custom_target_tasks.py
index e5fea7bf6..5b5423e51 100644
--- a/taskcluster/self_taskgraph/custom_target_tasks.py
+++ b/taskcluster/self_taskgraph/custom_target_tasks.py
@@ -9,7 +9,7 @@ def target_tasks_release(full_task_graph, parameters, graph_config):
 
     # Ensure we don't run `taskcluster-taskgraph` release tasks when publishing
     # `pytest-taskgraph`.
-    head_ref = parameters.rsplit("/", 1)[-1]
+    head_ref = parameters["head_ref"].rsplit("/", 1)[-1]
     if head_ref.startswith("pytest-taskgraph"):
         tasks = {
             l: t