diff --git a/src/test/regress/expected/failure_connection_establishment.out b/src/test/regress/expected/failure_connection_establishment.out
index 49f159397..2cab5c695 100644
--- a/src/test/regress/expected/failure_connection_establishment.out
+++ b/src/test/regress/expected/failure_connection_establishment.out
@@ -231,6 +231,7 @@ SELECT citus.mitmproxy('conn.allow()');
 
 (1 row)
 
+SET citus.enable_stat_counters TO true;
 SET citus.force_max_query_parallelization TO ON;
 SET citus.node_connection_timeout TO 900;
 SELECT citus.mitmproxy('conn.connect_delay(1400)');
@@ -265,6 +266,7 @@ FROM pg_database WHERE datname = current_database();
 
 RESET citus.force_max_query_parallelization;
 RESET citus.node_connection_timeout;
+RESET citus.enable_stat_counters;
 SELECT citus.mitmproxy('conn.allow()');
  mitmproxy
 ---------------------------------------------------------------------
diff --git a/src/test/regress/sql/failure_connection_establishment.sql b/src/test/regress/sql/failure_connection_establishment.sql
index aedc1e798..e516cad4c 100644
--- a/src/test/regress/sql/failure_connection_establishment.sql
+++ b/src/test/regress/sql/failure_connection_establishment.sql
@@ -124,6 +124,7 @@ RESET citus.force_max_query_parallelization;
 RESET citus.node_connection_timeout;
 SELECT citus.mitmproxy('conn.allow()');
 
+SET citus.enable_stat_counters TO true;
 SET citus.force_max_query_parallelization TO ON;
 SET citus.node_connection_timeout TO 900;
 SELECT citus.mitmproxy('conn.connect_delay(1400)');
@@ -148,6 +149,7 @@ FROM pg_database WHERE datname = current_database();
 
 RESET citus.force_max_query_parallelization;
 RESET citus.node_connection_timeout;
+RESET citus.enable_stat_counters;
 SELECT citus.mitmproxy('conn.allow()');
 
 -- one similar test, and this time on modification queries