diff --git a/tests/test_client_v2.py b/tests/test_client_v2.py
index 235fc9436..fcee54981 100644
--- a/tests/test_client_v2.py
+++ b/tests/test_client_v2.py
@@ -14,7 +14,7 @@
 class TestClientV2(unittest.TestCase):
 
     def test_chat(self) -> None:
-        response = co.chat(model="command-r-plus", messages=[cohere.UserChatMessageV2(content="hello world!")])
+        response = co.chat(model="command-r-plus", documents=["yoyo"], messages=[cohere.UserChatMessageV2(content="hello world!")])
 
         print(response.message)