diff --git a/src/test/regress/expected/citus_drain_node.out b/src/test/regress/expected/citus_drain_node.out
index 1957535b4..647cbe38e 100644
--- a/src/test/regress/expected/citus_drain_node.out
+++ b/src/test/regress/expected/citus_drain_node.out
@@ -115,6 +115,13 @@ SELECT nodename, nodeport, COUNT(*)
  localhost |    57638 |     2
 (2 rows)
 
+-- flakyness test
+SELECT floor(random() * 2) + 1;
+ ?column?
+---------------------------------------------------------------------
+        2
+(1 row)
+
 RESET search_path;
 SET client_min_messages TO WARNING;
 DROP SCHEMA citus_drain_node CASCADE;
diff --git a/src/test/regress/sql/citus_drain_node.sql b/src/test/regress/sql/citus_drain_node.sql
index c105a0245..5f898572e 100644
--- a/src/test/regress/sql/citus_drain_node.sql
+++ b/src/test/regress/sql/citus_drain_node.sql
@@ -56,6 +56,9 @@ SELECT nodename, nodeport, COUNT(*)
  WHERE placement.groupid = node.groupid
    AND node.noderole = 'primary' GROUP BY nodename, nodeport ORDER BY 1,2;
 
+-- flakyness test
+SELECT floor(random() * 2) + 1;
+
 RESET search_path;
 
 SET client_min_messages TO WARNING;