diff --git a/tsl/test/expected/bgw_db_scheduler.out b/tsl/test/expected/bgw_db_scheduler.out
index 88d906dbf..6a316b4c1 100644
--- a/tsl/test/expected/bgw_db_scheduler.out
+++ b/tsl/test/expected/bgw_db_scheduler.out
@@ -276,6 +276,13 @@ SELECT application_name FROM pg_stat_activity WHERE application_name LIKE 'User-
 ------------------
 (0 rows)
 
+-- wait for scheduler finish
+SELECT ts_bgw_db_scheduler_test_wait_for_scheduler_finish();
+ ts_bgw_db_scheduler_test_wait_for_scheduler_finish 
+----------------------------------------------------
+ 
+(1 row)
+
 --
 -- Test running a normal job
 --
diff --git a/tsl/test/sql/bgw_db_scheduler.sql b/tsl/test/sql/bgw_db_scheduler.sql
index c268461aa..9e78ba009 100644
--- a/tsl/test/sql/bgw_db_scheduler.sql
+++ b/tsl/test/sql/bgw_db_scheduler.sql
@@ -185,6 +185,9 @@ SELECT delete_job(:job_id);
 RESET client_min_messages;
 SELECT application_name FROM pg_stat_activity WHERE application_name LIKE 'User-Defined Action%';
 
+-- wait for scheduler finish
+SELECT ts_bgw_db_scheduler_test_wait_for_scheduler_finish();
+
 --
 -- Test running a normal job
 --