diff --git a/pointnet2/pointnet2_utils.py b/pointnet2/pointnet2_utils.py
index 7ade909..4f617fe 100755
--- a/pointnet2/pointnet2_utils.py
+++ b/pointnet2/pointnet2_utils.py
@@ -388,6 +388,7 @@ def __init__(self, use_xyz=True, ret_grouped_xyz=False):
         # type: (GroupAll, bool) -> None
         super(GroupAll, self).__init__()
         self.use_xyz = use_xyz
+        self.ret_grouped_xyz = ret_grouped_xyz
 
     def forward(self, xyz, new_xyz, features=None):
         # type: (GroupAll, torch.Tensor, torch.Tensor, torch.Tensor) -> Tuple[torch.Tensor]