diff --git a/src/llm/llm.py b/src/llm/llm.py
index 24f20d59..55d210ae 100644
--- a/src/llm/llm.py
+++ b/src/llm/llm.py
@@ -68,7 +68,7 @@ def __init__(self, model_id: str = None):
             
         }
         if ollama.client:
-            self.models["OLLAMA"] = [(model["name"], model["name"]) for model in ollama.models]
+            self.models["OLLAMA"] = [(model.model, model.model) for model in ollama.models]
 
     def list_models(self) -> dict:
         return self.models