diff --git a/tests/local_testing/test_completion.py b/tests/local_testing/test_completion.py
index c024b43e3c1..582f219a78c 100644
--- a/tests/local_testing/test_completion.py
+++ b/tests/local_testing/test_completion.py
@@ -4140,7 +4140,7 @@ async def test_completion_ai21_chat():
     litellm.set_verbose = True
     try:
         response = await litellm.acompletion(
-            model="jamba-1.5-large",
+            model="jamba-mini",
             user="ishaan",
             tool_choice="auto",
             seed=123,
diff --git a/tests/local_testing/test_streaming.py b/tests/local_testing/test_streaming.py
index aee72e5b283..9ac2d86c003 100644
--- a/tests/local_testing/test_streaming.py
+++ b/tests/local_testing/test_streaming.py
@@ -553,7 +553,7 @@ async def test_completion_predibase_streaming(sync_mode):
 async def test_completion_ai21_stream():
     litellm.set_verbose = True
     response = await litellm.acompletion(
-        model="ai21_chat/jamba-1.5-large",
+        model="ai21_chat/jamba-mini",
         user="ishaan",
         stream=True,
         seed=123,