diff --git a/models/pointnet2_msg_sem.py b/models/pointnet2_msg_sem.py
index b7a00f3..c59de03 100644
--- a/models/pointnet2_msg_sem.py
+++ b/models/pointnet2_msg_sem.py
@@ -62,7 +62,6 @@ class Pointnet2MSG(nn.Module):
                 radii=[0.1, 0.2],
                 nsamples=[16, 32],
                 mlps=[[c_in, 64, 64, 128], [c_in, 64, 96, 128]],
-                # use_xyz=use_xyz
             )
         )
         c_out_1 = 128 + 128
@@ -74,7 +73,6 @@ class Pointnet2MSG(nn.Module):
                 radii=[0.2, 0.4],
                 nsamples=[16, 32],
                 mlps=[[c_in, 128, 196, 256], [c_in, 128, 196, 256]],
-                # use_xyz=use_xyz
             )
         )
         c_out_2 = 256 + 256
@@ -86,7 +84,6 @@ class Pointnet2MSG(nn.Module):
                 radii=[0.4, 0.8],
                 nsamples=[16, 32],
                 mlps=[[c_in, 256, 256, 512], [c_in, 256, 384, 512]],
-                # use_xyz=use_xyz
             )
         )
         c_out_3 = 512 + 512
diff --git a/utils/pointnet2_modules.py b/utils/pointnet2_modules.py
index bce9154..eb866d5 100644
--- a/utils/pointnet2_modules.py
+++ b/utils/pointnet2_modules.py
@@ -22,7 +22,7 @@ class _PointnetSAModuleBase(nn.Module):
         ----------
         xyz : torch.Tensor
             (B, N, 3) tensor of the xyz coordinates of the points
-        point : torch.Tensor
+        points : torch.Tensor
             (B, N, C) tensor of the descriptors of the the points
 
         Returns