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