diff --git a/tests/python_client/base/client_base.py b/tests/python_client/base/client_base.py
index 3e87447cdd..e566f21754 100644
--- a/tests/python_client/base/client_base.py
+++ b/tests/python_client/base/client_base.py
@@ -196,7 +196,7 @@ class TestcaseBase(Base):
         collection_w.insert(df_default)
         conn.flush([collection_w.name])
         collection_w.load(partition_names=[partition_w.name, "_default"])
-        return collection_w, partition_w,df_partition, df_default
+        return collection_w, partition_w, df_partition, df_default
    
 
     def insert_entities_into_two_partitions(self, half, prefix='query'):