diff --git a/src/backend/distributed/sql/citus--11.1-1--11.2-1.sql b/src/backend/distributed/sql/citus--11.1-1--11.2-1.sql
index d5bfb4184..f7854602a 100644
--- a/src/backend/distributed/sql/citus--11.1-1--11.2-1.sql
+++ b/src/backend/distributed/sql/citus--11.1-1--11.2-1.sql
@@ -1 +1,2 @@
 #include "udfs/get_rebalance_progress/11.2-1.sql"
+#include "udfs/citus_isolation_test_session_is_blocked/11.2-1.sql"
diff --git a/src/backend/distributed/sql/downgrades/citus--11.2-1--11.1-1.sql b/src/backend/distributed/sql/downgrades/citus--11.2-1--11.1-1.sql
index b2e6c1e2c..455780d45 100644
--- a/src/backend/distributed/sql/downgrades/citus--11.2-1--11.1-1.sql
+++ b/src/backend/distributed/sql/downgrades/citus--11.2-1--11.1-1.sql
@@ -1 +1,2 @@
 #include "../udfs/get_rebalance_progress/11.1-1.sql"
+#include "../udfs/citus_isolation_test_session_is_blocked/11.1-1.sql"