diff --git a/python/cugraph-pyg/cugraph_pyg/loader/node_loader.py b/python/cugraph-pyg/cugraph_pyg/loader/node_loader.py
index 5536c7654de..6b82c68ff44 100644
--- a/python/cugraph-pyg/cugraph_pyg/loader/node_loader.py
+++ b/python/cugraph-pyg/cugraph_pyg/loader/node_loader.py
@@ -136,6 +136,7 @@ def __init__(
         self.__batch_size = batch_size
         self.__shuffle = shuffle
         self.__drop_last = drop_last
+        self.__global_shuffle = global_shuffle
 
     def __get_input(self):
         _, graph_store = self.__data