diff --git a/tests/test_ai_cache_tracking.py b/tests/test_ai_cache_tracking.py
index c9d508fa..3e7171a3 100644
--- a/tests/test_ai_cache_tracking.py
+++ b/tests/test_ai_cache_tracking.py
@@ -1,6 +1,7 @@
 import unittest
 from unittest.mock import patch, MagicMock
 from src import ai_client
+from src.result_types import Result
 import time
 
 def test_gemini_cache_tracking() -> None:
@@ -44,11 +45,12 @@ def test_gemini_cache_tracking() -> None:
    mock_client.caches.list.return_value = [MagicMock(size_bytes=5000)]
    
    # Act
-   ai_client.send(
+   result = ai_client.send_result(
     md_content="Some long context that triggers caching",
     user_message="Hello",
     file_items=file_items
    )
+   assert result.ok
    
    # Assert
    stats = ai_client.get_gemini_cache_stats()