diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py
index b22e69f0942..3b2bc7e647d 100644
--- a/tests/test_litellm/test_utils.py
+++ b/tests/test_litellm/test_utils.py
@@ -4432,7 +4432,7 @@ _FIREWORKS_MODELS = [
         4e-06,
         1.9e-07,
         262144,
-        262144,
+        32768,
         True,
         True,
     ),
@@ -4442,7 +4442,7 @@ _FIREWORKS_MODELS = [
         8e-06,
         3.8e-07,
         262144,
-        262144,
+        32768,
         True,
         True,
     ),
@@ -4452,7 +4452,7 @@ _FIREWORKS_MODELS = [
         4e-06,
         1.6e-07,
         262144,
-        262144,
+        32768,
         True,
         True,
     ),
@@ -4462,7 +4462,7 @@ _FIREWORKS_MODELS = [
         8e-06,
         3e-07,
         262144,
-        262144,
+        32768,
         True,
         True,
     ),