diff --git a/src/backends/dlb_talp_tree/dlb_talp_tree.cpp b/src/backends/dlb_talp_tree/dlb_talp_tree.cpp
index 50802a7a27eff0ac652c56a4f4d1e525599784e0..854ce39b37588bcbdafc2097083b786eb45925be 100644
--- a/src/backends/dlb_talp_tree/dlb_talp_tree.cpp
+++ b/src/backends/dlb_talp_tree/dlb_talp_tree.cpp
@@ -121,8 +121,8 @@ void DLBTalpTreeStrategy::Finalize() noexcept {
     counts.resize(mpi_size);
     displs.resize(mpi_size);
 
-    MPI_Allgather(&num_regions, 1, MPI_INT, counts.data(), 1, MPI_INT,
-                  MPI_COMM_WORLD);
+    PMPI_Allgather(&num_regions, 1, MPI_INT, counts.data(), 1, MPI_INT,
+                   MPI_COMM_WORLD);
 
     int num_all_regions = 0;
     for (int i = 0; i < counts.size(); i++) {