diff --git a/utils/datasets.py b/utils/datasets.py
index 52b028994325..25a2ba6f9561 100755
--- a/utils/datasets.py
+++ b/utils/datasets.py
@@ -568,6 +568,7 @@ def __getitem__(self, index):
         if self.augment:
             # Albumentations
             img, labels = self.albumentations(img, labels)
+            nl = len(labels) # update after albumentations
 
             # HSV color-space
             augment_hsv(img, hgain=hyp['hsv_h'], sgain=hyp['hsv_s'], vgain=hyp['hsv_v'])