diff --git a/tf_compare.py b/tf_compare.py
index 1469e29..e5e459f 100644
--- a/tf_compare.py
+++ b/tf_compare.py
@@ -63,7 +63,7 @@ def get_workflow_runs():
 
         workflow_runs = artifact_data.get("workflow_runs", [])
 
-        branch = workflow_runs[0]['pull_requests'][0]['base']['ref']
+        branch = workflow_runs[0]['head_branch']
 
         # download workflow run log for new run that is currently in progress
         allowed_workflows = {'Beta', 'Monthly', 'Red'}