From 9dfafa743cf56b9f36fc0df70f2f3bf3d15dea03 Mon Sep 17 00:00:00 2001
From: Baizhou Zhang <sobereddiezhang@gmail.com>
Date: Sun, 9 Mar 2025 12:45:39 -0700
Subject: [PATCH] Fix test of flashinfer mla with nextn (#4237)

---
 test/srt/test_mla_flashinfer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/srt/test_mla_flashinfer.py b/test/srt/test_mla_flashinfer.py
index e7113d03d..6f17c6ff9 100644
--- a/test/srt/test_mla_flashinfer.py
+++ b/test/srt/test_mla_flashinfer.py
@@ -121,7 +121,7 @@ class TestFlashinferMLAMTP(unittest.TestCase):
                     "--speculative-draft",
                     "lmsys/sglang-ci-dsv3-test-NextN",
                     "--speculative-num-steps",
-                    "4",
+                    "3",
                     "--speculative-eagle-topk",
                     "1",
                     "--speculative-num-draft-tokens",