diff --git a/pina/label_tensor.py b/pina/label_tensor.py
index cce141c..d94943b 100644
--- a/pina/label_tensor.py
+++ b/pina/label_tensor.py
@@ -479,9 +479,10 @@ class LabelTensor(torch.Tensor):
 
         # Retrieve selected tensor and labels
         selected_tensor = super().__getitem__(index)
-        if not hasattr(self, "_labels"):
+        if hasattr(self, "_labels"):
+            original_labels=self._labels
+        else:
             return selected_tensor
-
         original_labels = self._labels
         updated_labels = copy(original_labels)
 
diff --git a/pina/model/block/convolution_2d.py b/pina/model/block/convolution_2d.py
index 68df175..61f594e 100644
--- a/pina/model/block/convolution_2d.py
+++ b/pina/model/block/convolution_2d.py
@@ -500,6 +500,7 @@ class ContinuousConvBlock(BaseContinuousConv):
 
         # initialize grid
         X = self._grid_transpose.clone().detach()
+        
         conv_transposed = self._grid_transpose.clone().detach()
 
         # list to iterate for calculating nn output