diff --git a/tests/python_client/deploy/scripts/action_after_reinstall.py b/tests/python_client/deploy/scripts/action_after_reinstall.py
index 08d251f999..00a2ef2c56 100644
--- a/tests/python_client/deploy/scripts/action_after_reinstall.py
+++ b/tests/python_client/deploy/scripts/action_after_reinstall.py
@@ -40,5 +40,5 @@ if __name__ == '__main__':
     host = args.host 
     data_size = args.data_size
     print(f"data size: {data_size}")
-    task_1(data_size)
-    task_2(data_size)
\ No newline at end of file
+    task_1(data_size, host)
+    task_2(data_size, host)
\ No newline at end of file