diff --git a/python/gigl/distributed/dist_partitioner.py b/python/gigl/distributed/dist_partitioner.py
index f7b3d930..821dc76d 100644
--- a/python/gigl/distributed/dist_partitioner.py
+++ b/python/gigl/distributed/dist_partitioner.py
@@ -1678,6 +1678,7 @@ def partition(
             )
         else:
             partitioned_node_features = None
+            partitioned_node_labels = None
 
         if self._positive_label_edge_index is not None:
             partitioned_positive_edge_index = self.partition_labels(