diff --git a/tests/test_rag_integration.py b/tests/test_rag_integration.py
index b552f70c..2c200c10 100644
--- a/tests/test_rag_integration.py
+++ b/tests/test_rag_integration.py
@@ -8,6 +8,7 @@ from src.app_controller import AppController
 from src import ai_client
 from src import events
 from src import models
+from src.result_types import Result
 
 @pytest.fixture
 def mock_project():
@@ -74,7 +75,7 @@ def test_rag_integration(mock_project):
  # _send_gemini which is what actually "sends to the provider".
  with patch('src.ai_client.send_result', wraps=ai_client.send_result) as mock_send:
   with patch('src.ai_client._send_gemini') as mock_provider:
-   mock_provider.return_value = "Mock AI Response"
+   mock_provider.return_value = Result(data="Mock AI Response")
    
    # Create a UserRequestEvent as if the user clicked "Gen + Send"
    event = events.UserRequestEvent(