diff --git a/ultralytics/data/base.py b/ultralytics/data/base.py
index ddfce6eed..6af8d3cc4 100644
--- a/ultralytics/data/base.py
+++ b/ultralytics/data/base.py
@@ -120,7 +120,9 @@ class BaseDataset(Dataset):
         except Exception as e:
             raise FileNotFoundError(f"{self.prefix}Error loading data from {img_path}\n{HELP_URL}") from e
         if self.fraction < 1:
-            im_files = im_files[: round(len(im_files) * self.fraction)]
+            # im_files = im_files[: round(len(im_files) * self.fraction)]
+            num_elements_to_select = round(len(im_files) * self.fraction)
+            im_files = random.sample(im_files, num_elements_to_select)
         return im_files
 
     def update_labels(self, include_class: Optional[list]):