diff --git a/src/backend/distributed/transaction/transaction_management.c b/src/backend/distributed/transaction/transaction_management.c
index e78563fc6..b71694439 100644
--- a/src/backend/distributed/transaction/transaction_management.c
+++ b/src/backend/distributed/transaction/transaction_management.c
@@ -27,6 +27,7 @@
 #include "distributed/transaction_management.h"
 #include "distributed/placement_connection.h"
 #include "distributed/subplan_execution.h"
+#include "distributed/version_compat.h"
 #include "utils/hsearch.h"
 #include "utils/guc.h"
 #include "utils/memutils.h"
@@ -204,7 +205,7 @@ CoordinatedTransactionCallback(XactEvent event, void *arg)
 				 * been closed; force them closed here before running
 				 * RemoveIntermediateResultsDirectory.
 				 */
-				AtEOXact_Files();
+				AtEOXact_Files(false);
 				SwallowErrors(RemoveIntermediateResultsDirectory);
 			}
 			ResetShardPlacementTransactionState();
diff --git a/src/backend/distributed/worker/task_tracker.c b/src/backend/distributed/worker/task_tracker.c
index d2543745a..a1bde5d94 100644
--- a/src/backend/distributed/worker/task_tracker.c
+++ b/src/backend/distributed/worker/task_tracker.c
@@ -179,7 +179,7 @@ TaskTrackerMain(Datum main_arg)
 		 * hash and an LWLock guarding that hash.
 		 */
 		LWLockReleaseAll();
-		AtEOXact_Files();
+		AtEOXact_Files(false);
 		AtEOXact_HashTables(false);
 
 		/*
diff --git a/src/include/distributed/version_compat.h b/src/include/distributed/version_compat.h
index a638027f9..8c97ca684 100644
--- a/src/include/distributed/version_compat.h
+++ b/src/include/distributed/version_compat.h
@@ -55,6 +55,8 @@
 #define BackgroundWorkerInitializeConnectionByOid(dboid, useroid, flags) \
 	BackgroundWorkerInitializeConnectionByOid(dboid, useroid)
 
+#define AtEOXact_Files(isCommit) \
+	AtEOXact_Files()
 
 #define ACLCHECK_OBJECT_TABLE ACL_KIND_CLASS
 #define ACLCHECK_OBJECT_SCHEMA ACL_KIND_NAMESPACE