diff --git a/tests/python_client/testcases/test_insert_20.py b/tests/python_client/testcases/test_insert_20.py
index b1fa6d1eb8..687c9b5c6f 100644
--- a/tests/python_client/testcases/test_insert_20.py
+++ b/tests/python_client/testcases/test_insert_20.py
@@ -721,7 +721,7 @@ class TestInsertAsync(TestcaseBase):
         expected: verify num entities
         """
         collection_w = self.init_collection_wrap(name=cf.gen_unique_str(prefix))
-        df = cf.gen_default_dataframe_data(nb=ct.default_nb)
+        df = cf.gen_default_dataframe_data()
         future, _ = collection_w.insert(data=df, _async=True)
         future.done()
         mutation_res = future.result()
@@ -737,7 +737,7 @@ class TestInsertAsync(TestcaseBase):
         expected: verify num entities
         """
         collection_w = self.init_collection_wrap(name=cf.gen_unique_str(prefix))
-        df = cf.gen_default_dataframe_data(nb=ct.default_nb)
+        df = cf.gen_default_dataframe_data()
         mutation_res, _ = collection_w.insert(data=df, _async=False)
         assert mutation_res.insert_count == ct.default_nb
         assert mutation_res.primary_keys == df[ct.default_int64_field_name].values.tolist()
@@ -751,7 +751,7 @@ class TestInsertAsync(TestcaseBase):
         expected: verify num entities
         """
         collection_w = self.init_collection_wrap(name=cf.gen_unique_str(prefix))
-        df = cf.gen_default_dataframe_data(nb=ct.default_nb)
+        df = cf.gen_default_dataframe_data()
         future, _ = collection_w.insert(data=df, _async=True, _callback=assert_mutation_result)
         future.done()
         mutation_res = future.result()