diff --git a/tests/python_client/testcases/test_search_20.py b/tests/python_client/testcases/test_search_20.py
index ed61744f71..4cfcd03f0d 100644
--- a/tests/python_client/testcases/test_search_20.py
+++ b/tests/python_client/testcases/test_search_20.py
@@ -1081,26 +1081,27 @@ class TestCollectionSearch(TestcaseBase):
 
     @pytest.mark.tags(CaseLabel.L2)
     #@pytest.mark.skip(reason="skip temporarily for debug")
-    def test_search_max_dim(self, nq, auto_id, _async):
+    def test_search_max_dim(self, auto_id, _async):
         """
         target: test search with max configuration
         method: create connection, collection, insert and search with max dim
         expected: search successfully with limit(topK)
         """
         # 1. initialize with data
-        collection_w, _, _, insert_ids = self.init_collection_general(prefix, True, default_nb,
+        collection_w, _, _, insert_ids = self.init_collection_general(prefix, True, 100,
                                                                       auto_id=auto_id,
                                                                       dim=max_dim)
         # 2. search
+        nq = 2
         log.info("test_search_max_dim: searching collection %s" % collection_w.name)
         vectors = [[random.random() for _ in range(max_dim)] for _ in range(nq)]
         collection_w.search(vectors[:nq], default_search_field,
-                            default_search_params, 2,
+                            default_search_params, nq,
                             default_search_exp, _async=_async,
                             check_task=CheckTasks.check_search_results,
                             check_items={"nq": nq,
                                          "ids": insert_ids,
-                                         "limit": 2,
+                                         "limit": nq,
                                          "_async": _async})
 
     @pytest.mark.tags(CaseLabel.L2)