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']}"