diff --git a/tests/test_gemini_cli_edge_cases.py b/tests/test_gemini_cli_edge_cases.py
index 253ccde9..7cba8455 100644
--- a/tests/test_gemini_cli_edge_cases.py
+++ b/tests/test_gemini_cli_edge_cases.py
@@ -1,6 +1,7 @@
 from unittest.mock import patch, MagicMock
 from src.gemini_cli_adapter import GeminiCliAdapter
 from src import mcp_client
+from src.result_types import Result
 
 def test_gemini_cli_context_bleed_prevention() -> None:
  import src.ai_client as ai_client
@@ -35,5 +36,6 @@ def test_gemini_cli_loop_termination() -> None:
   mock_process.returncode = 0
   mock_popen.return_value = mock_process
   ai_client.set_provider("gemini_cli", "gemini-2.0-flash")
-  result = ai_client.send("context", "prompt")
-  assert result == "Final answer"
+  result = ai_client.send_result("context", "prompt")
+  assert result.ok
+  assert result.data == "Final answer"