diff --git a/shellgenius/ollama_client.py b/shellgenius/ollama_client.py
index fc99748..733aaf6 100644
--- a/shellgenius/ollama_client.py
+++ b/shellgenius/ollama_client.py
@@ -21,8 +21,8 @@ class OllamaClient:
             model: Model name to use
         """
         config = get_config()
-        self.host = host or config.get("ollama_host", "localhost:11434")
-        self.model = model or config.get("ollama_model", "llama3")
+        self.host = host or config.ollama_host
+        self.model = model or config.ollama_model
         self._client: Optional[ollama.Client] = None
 
     @property