diff --git a/shellgenius/ollama_client.py b/shellgenius/ollama_client.py
index 9d693b8..8da7f19 100644
--- a/shellgenius/ollama_client.py
+++ b/shellgenius/ollama_client.py
@@ -1,6 +1,5 @@
 """Ollama client wrapper for ShellGenius."""
 
-import json
 import logging
 from typing import Any, Dict, Generator, List, Optional
 
@@ -96,7 +95,7 @@ class OllamaClient:
         """
         model = model or self.model
         try:
-            response = self.client.generate(
+            response = self.client.generate(  # type: ignore[call-overload]
                 model=model,
                 prompt=prompt,
                 stream=stream,
@@ -147,12 +146,13 @@ class OllamaClient:
         """
         model = model or self.model
         try:
-            response = self.client.chat(
+            response = self.client.chat(  # type: ignore[call-overload]
                 model=model,
                 messages=messages,
                 stream=stream,
             )
             return {"success": True, "response": response}
+            return {"success": True, "response": response}
         except Exception as e:
             logger.error(f"Chat failed: {e}")
             return {"success": False, "error": str(e), "response": None}