diff --git a/tests/python_client/deploy/scripts/action_after_reinstall.py b/tests/python_client/deploy/scripts/action_after_reinstall.py
index 2ac9bf18b8..df490bb830 100644
--- a/tests/python_client/deploy/scripts/action_after_reinstall.py
+++ b/tests/python_client/deploy/scripts/action_after_reinstall.py
@@ -5,7 +5,7 @@ from utils import *
 def task_1():
     """
     task_1:
-        before reinstall: create collection and insert data
+        before reinstall: create collection and insert data, load and search
         after reinstall: get collection, load, search, create index, load, and search
     """
     prefix = "task_1_"
@@ -18,7 +18,7 @@ def task_1():
 def task_2():
     """
     task_2:
-        before reinstall: create collection, insert data and create index
+        before reinstall: create collection, insert data and create index, load and search
         after reinstall: get collection, load, search, insert data, create index, load, and search
     """
     prefix = "task_2_"
diff --git a/tests/python_client/deploy/scripts/action_after_upgrade.py b/tests/python_client/deploy/scripts/action_after_upgrade.py
index 52adf56c84..e664812501 100644
--- a/tests/python_client/deploy/scripts/action_after_upgrade.py
+++ b/tests/python_client/deploy/scripts/action_after_upgrade.py
@@ -5,7 +5,7 @@ from utils import *
 def task_1():
     """
     task_1:
-        before upgrade: create collection and insert data
+        before upgrade: create collection and insert data, load and search
         after upgrade: get collection, load, search, create index, load, and search
     """
     prefix = "task_1_"
@@ -13,12 +13,14 @@ def task_1():
     get_collections(prefix)
     load_and_search(prefix)
     create_collections_and_insert_data(prefix)
+    create_index(prefix)
+    load_and_search(prefix)
 
 
 def task_2():
     """
     task_2:
-        before upgrade: create collection, insert data and create index
+        before upgrade: create collection, insert data and create index, load and search
         after upgrade: get collection, load, search, insert data, create index, load, and search
     """
     prefix = "task_2_"
diff --git a/tests/python_client/deploy/scripts/action_before_reinstall.py b/tests/python_client/deploy/scripts/action_before_reinstall.py
index 766cde0586..842fdb0b04 100644
--- a/tests/python_client/deploy/scripts/action_before_reinstall.py
+++ b/tests/python_client/deploy/scripts/action_before_reinstall.py
@@ -5,7 +5,7 @@ from utils import *
 def task_1():
     """
     task_1:
-        before reinstall: create collection and insert data
+        before reinstall: create collection and insert data, load and search
         after reinstall: get collection, load, search, create index, load, and search
     """
     prefix = "task_1_"
@@ -13,12 +13,13 @@ def task_1():
     get_collections(prefix)
     load_and_search(prefix)
     create_collections_and_insert_data(prefix)
+    load_and_search(prefix)
 
 
 def task_2():
     """
     task_2:
-        before reinstall: create collection, insert data and create index
+        before reinstall: create collection, insert data and create index,load and search
         after reinstall: get collection, load, search, insert data, create index, load, and search
     """
     prefix = "task_2_"
diff --git a/tests/python_client/deploy/scripts/action_before_upgrade.py b/tests/python_client/deploy/scripts/action_before_upgrade.py
index 70f834a178..b151faa840 100644
--- a/tests/python_client/deploy/scripts/action_before_upgrade.py
+++ b/tests/python_client/deploy/scripts/action_before_upgrade.py
@@ -5,7 +5,7 @@ from utils import *
 def task_1():
     """
     task_1:
-        before upgrade: create collection and insert data
+        before upgrade: create collection and insert data, load and search
         after upgrade: get collection, load, search, create index, load, and search
     """
     prefix = "task_1_"
@@ -13,12 +13,13 @@ def task_1():
     get_collections(prefix)
     load_and_search(prefix)
     create_collections_and_insert_data(prefix)
+    load_and_search(prefix)
 
 
 def task_2():
     """
     task_2:
-        before upgrade: create collection, insert data and create index
+        before upgrade: create collection, insert data and create index, load and search
         after upgrade: get collection, load, search, insert data, create index, load, and search
     """
     prefix = "task_2_"