diff --git a/src/datasets/segment3d.py b/src/datasets/segment3d.py
index 49e2e3c..90db76d 100644
--- a/src/datasets/segment3d.py
+++ b/src/datasets/segment3d.py
@@ -216,7 +216,7 @@ def all_base_cloud_ids(self):
             `{'train': [...], 'val': [...], 'test': [...]}`
         """
         return    {
-            'train': ['DrumPlant_Tank_rotate45_classified4.las',"DrumPlant_Tank_rotate45_classified8.las"],
+            'train': ['DrumPlant_Tank_rotate45_classified4.las',"DrumPlant_Tank_rotate45_classified8.las","DrumPlant_Tank_rotate45_classified.las"],
             'val': ['DrumPlant_Tank_rotate45_classified_sample2.las', 'DrumPlant_Tank_rotate45_classified_sample1.las'],
             'test':  ['DrumPlant_Tank_rotate45_classified_small.las'] 
             }