From 1a693f31b592e92f709f63dda1a6370cf1a34c11 Mon Sep 17 00:00:00 2001
From: jonathanmetzman <31354670+jonathanmetzman@users.noreply.github.com>
Date: Sat, 14 Sep 2024 23:10:17 -0400
Subject: [PATCH] Use correct proto field name (#4240)

---
 src/clusterfuzz/_internal/bot/untrusted_runner/tasks_host.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/clusterfuzz/_internal/bot/untrusted_runner/tasks_host.py b/src/clusterfuzz/_internal/bot/untrusted_runner/tasks_host.py
index cc6e932d55..3ad27b3784 100644
--- a/src/clusterfuzz/_internal/bot/untrusted_runner/tasks_host.py
+++ b/src/clusterfuzz/_internal/bot/untrusted_runner/tasks_host.py
@@ -14,6 +14,7 @@
 """Tasks host."""
 
 import datetime
+import os
 
 from google.protobuf import wrappers_pb2
 import grpc
@@ -86,7 +87,7 @@ def do_corpus_pruning(uworker_input, context, revision):
           crash_type=crash.crash_type,
           crash_address=crash.crash_address,
           crash_stacktrace=crash.crash_stacktrace,
-          unit_path=crash.unit_path,
+          unit_name=os.path.basename(crash.unit_path),
           security_flag=crash.security_flag,
       ) for crash in response.crashes
   ]