diff --git a/lib/queue/worker.py b/lib/queue/worker.py
index 3b0f2df..b45bbcb 100644
--- a/lib/queue/worker.py
+++ b/lib/queue/worker.py
@@ -52,7 +52,7 @@ def safely_respond(self, model: Model) -> List[schemas.Message]:
         if messages_with_queues:
             logger.debug(f"About to respond to: ({messages_with_queues})")
             try:
-                responses = model.respond([schemas.Message(**{**json.loads(message.body), **{"model_name": self.model.model_name}}) for message, queue in messages_with_queues])
+                responses = model.respond([schemas.Message(**{**json.loads(message.body), **{"model_name": model.model_name}}) for message, queue in messages_with_queues])
             except Exception as e:
                 logger.error(e)
             self.delete_messages(messages_with_queues)