diff --git a/python/cugraph-dgl/cugraph_dgl/features.py b/python/cugraph-dgl/cugraph_dgl/features.py
index b4ff0049494..9dc009f4127 100644
--- a/python/cugraph-dgl/cugraph_dgl/features.py
+++ b/python/cugraph-dgl/cugraph_dgl/features.py
@@ -51,7 +51,7 @@ def __init__(
         location: str(optional, default='cpu')
             The location ('cpu' or 'cuda') where data is stored.
         """
-        self.__wg_comm = wgth.get_local_node_communicator()
+        self.__wg_comm = wgth.get_global_communicator()
 
         if len(tensor.shape) > 2:
             raise ValueError("Only 1-D or 2-D tensors are supported by WholeGraph.")