diff --git a/utils/datasets.py b/utils/datasets.py
index 52b0289..25a2ba6 100755
--- a/utils/datasets.py
+++ b/utils/datasets.py
@@ -568,6 +568,7 @@ class LoadImagesAndLabels(Dataset):  # for training/testing
         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'])