diff --git a/torchgeo/datamodules/vhr10.py b/torchgeo/datamodules/vhr10.py
index 67a59b7ba80..abc2957b8ed 100644
--- a/torchgeo/datamodules/vhr10.py
+++ b/torchgeo/datamodules/vhr10.py
@@ -17,8 +17,6 @@
 from ..transforms import AugmentationSequential
 from .geo import NonGeoDataModule
 
-# kornia DataKeys must accept MASKS
-
 
 class _AugPipe(Module):
     """Pipeline for applying augmentations sequentially on select data keys."""
diff --git a/torchgeo/datasets/vhr10.py b/torchgeo/datasets/vhr10.py
index acee5989e77..7e62af59181 100644
--- a/torchgeo/datasets/vhr10.py
+++ b/torchgeo/datasets/vhr10.py
@@ -39,10 +39,7 @@ def convert_coco_poly_to_mask(
         mask = torch.as_tensor(mask, dtype=torch.uint8)
         mask = mask.any(dim=2)
         masks.append(mask)
-    if masks:
-        masks_tensor = torch.stack(masks, dim=0)
-    else:
-        masks_tensor = torch.zeros((0, height, width), dtype=torch.uint8)
+    masks_tensor = torch.stack(masks, dim=0)
     return masks_tensor