Fix failed tests

src/tests/integration_tests/test_entity_citation_validator.py

@@ -55,15 +55,14 @@ def test_entity_citation_succeeds_in_qa(
     llm = get_llm(
         __file__, test_entity_citation_succeeds_in_qa.__name__ + f"_{str(test_id)}"
     )
-    data = input_data[0]
 
     shared_memory = SharedMemory(
         objective=objective,
         agent_pool=None,
     )
 
     entity_validation = EntityValidation()
-    with patch.object(SearchTool, "_run", return_value=data):
+    with patch.object(SearchTool, "_run", return_value=input_data):
         task_agent = QAAgent(
             llm=llm,
             shared_memory=shared_memory,