diff --git a/src/test/regress/expected/failure_insert_select_repartition.out b/src/test/regress/expected/failure_insert_select_repartition.out
index 0cc9a5e86..c634cfff8 100644
--- a/src/test/regress/expected/failure_insert_select_repartition.out
+++ b/src/test/regress/expected/failure_insert_select_repartition.out
@@ -9,6 +9,7 @@ SELECT pg_backend_pid() as pid \gset
 SET citus.next_shard_id TO 4213581;
 SET citus.shard_count TO 4;
 SET citus.shard_replication_factor TO 2;
+SET citus.max_adaptive_executor_pool_size TO 1; -- have more consistent number of connections running
 CREATE TABLE replicated_source_table(a int, b int);
 SELECT create_distributed_table('replicated_source_table', 'a');
  create_distributed_table
@@ -69,9 +70,6 @@ SELECT citus.mitmproxy('conn.onQuery(query="worker_partition_query_result").kill
 (1 row)
 
 INSERT INTO target_table SELECT * FROM replicated_source_table;
-WARNING:  connection to the remote node localhost:xxxxx failed with the following error: server closed the connection unexpectedly
-	This probably means the server terminated abnormally
-	before or while processing the request.
 WARNING:  connection to the remote node localhost:xxxxx failed with the following error: server closed the connection unexpectedly
 	This probably means the server terminated abnormally
 	before or while processing the request.
diff --git a/src/test/regress/sql/failure_insert_select_repartition.sql b/src/test/regress/sql/failure_insert_select_repartition.sql
index 1f6eb1812..0c65c2529 100644
--- a/src/test/regress/sql/failure_insert_select_repartition.sql
+++ b/src/test/regress/sql/failure_insert_select_repartition.sql
@@ -12,6 +12,7 @@ SET citus.next_shard_id TO 4213581;
 
 SET citus.shard_count TO 4;
 SET citus.shard_replication_factor TO 2;
+SET citus.max_adaptive_executor_pool_size TO 1; -- have more consistent number of connections running
 CREATE TABLE replicated_source_table(a int, b int);
 SELECT create_distributed_table('replicated_source_table', 'a');
 INSERT INTO replicated_source_table SELECT i, i*i FROM generate_series(1, 10) i;