From 1b76fd1209bf68f3232d3dd07be8e69089c45e27 Mon Sep 17 00:00:00 2001
From: Morgan Epp <morgane@amazon.com>
Date: Fri, 1 Sep 2023 23:19:10 +0000
Subject: [PATCH] fix: use the proper entity key variable

---
 src/deadline_worker_agent/sessions/job_entities/job_entities.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/deadline_worker_agent/sessions/job_entities/job_entities.py b/src/deadline_worker_agent/sessions/job_entities/job_entities.py
index 747cb1e1..fb4d0cc4 100644
--- a/src/deadline_worker_agent/sessions/job_entities/job_entities.py
+++ b/src/deadline_worker_agent/sessions/job_entities/job_entities.py
@@ -267,7 +267,7 @@ def cache_entities(self, entity_identifiers: list[EntityIdentifier]):
                 # InternalServerException, ValidationException, ResourceNotFoundException,
 
                 failed_entity_key = self._entity_key(failed_entity)
-                entity_record = self._entity_record_map[entity_key]
+                entity_record = self._entity_record_map[failed_entity_key]
                 entity_record.error = failed_entity_value
                 logger.info(
                     f"BatchGetJobEntity request: {failed_entity_key} failed due to {failed_entity_value['code']}: {failed_entity_value['message']}"