diff --git a/tests/test_async_client.py b/tests/test_async_client.py
index 4350b4550..1198cec05 100644
--- a/tests/test_async_client.py
+++ b/tests/test_async_client.py
@@ -79,7 +79,9 @@ async def test_moved_fn(self) -> None:
     @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
     async def test_generate(self) -> None:
         response = await self.co.generate(
+            model="command-a-03-2025",
             prompt='Please explain to me how LLMs work',
+            max_tokens=10
         )
         print(response)