diff --git a/sCOCA_ML/dataset/momenta_dataset.py b/sCOCA_ML/dataset/momenta_dataset.py
index 5fdc6bb..a36b650 100644
--- a/sCOCA_ML/dataset/momenta_dataset.py
+++ b/sCOCA_ML/dataset/momenta_dataset.py
@@ -208,7 +208,7 @@ class MomentaDataset(Dataset):
             os.path.join(self.root_dir, self.INITIAL_CONDITIONS_DIR, f'ICs_{ID}_{var}.h5')
             for var in self.initial_conditions_variables
         ]
-        target_path = os.path.join(self.root_dir, self.TARGET_DIR, f'{self.target_variable}_{ID}_nforce{t}.h5')
+        target_path = os.path.join(self.root_dir, self.TARGET_DIR, f'{self.target_variable}_{ID}_{t}.h5')
         style_path = os.path.join(self.root_dir, self.STYLE_DIR, f'{self.style_files}_{ID}_nforce{t}.txt')
 
         # Read 3D chunks