diff --git a/tests/unit/rag/test_rag_query.py b/tests/unit/rag/test_rag_query.py
index 7b897bfe0..183b4d049 100644
--- a/tests/unit/rag/test_rag_query.py
+++ b/tests/unit/rag/test_rag_query.py
@@ -88,6 +88,7 @@ class TestRagQuery:
             config=MagicMock(),
             vector_io_api=MagicMock(),
             inference_api=MagicMock(),
+            files_api=MagicMock(),
         )
 
         vector_db_ids = ["db1", "db2"]