diff --git a/tests/python_client/testcases/test_compaction.py b/tests/python_client/testcases/test_compaction.py
index a594f664a0..6d14989f85 100644
--- a/tests/python_client/testcases/test_compaction.py
+++ b/tests/python_client/testcases/test_compaction.py
@@ -322,6 +322,9 @@ class TestCompactionParams(TestcaseBase):
 
         sleep(ct.max_compaction_interval + 1)
 
+        # create index
+        collection_w.create_index(ct.default_float_vec_field_name, ct.default_index)
+
         # verify queryNode load the compacted segments
         collection_w.load()
         replicas = collection_w.get_replicas()[0]
@@ -831,6 +834,7 @@ class TestCompactionOperation(TestcaseBase):
         collection_w.insert(df)
         assert collection_w.num_entities == tmp_nb
 
+        collection_w.create_index(ct.default_float_vec_field_name, ct.default_index)
         collection_w.load()
 
         seg_before, _ = self.utility_wrap.get_query_segment_info(collection_w.name)
@@ -1012,6 +1016,9 @@ class TestCompactionOperation(TestcaseBase):
         # create collection shard_num=1, insert 9 segments, each with one entity
         collection_w = self.collection_insert_multi_segments_one_shard(prefix, num_of_segment=less_threshold)
 
+        # create index
+        collection_w.create_index(ct.default_float_vec_field_name, ct.default_index)
+
         # load and verify no auto-merge
         collection_w.load()
         replicas = collection_w.get_replicas()[0]