diff --git a/Code/python/src/aiRNN/dataloader.py b/Code/python/src/aiRNN/dataloader.py
index a83ae74..2a6ed94 100644
--- a/Code/python/src/aiRNN/dataloader.py
+++ b/Code/python/src/aiRNN/dataloader.py
@@ -129,8 +129,8 @@ class SaveDataset(Dataset):
     def __init__(self, data_dict, step=1, start_offset=0, end_offset=0):
         super().__init__()
         self.step = step
-        self.start_offset = start_offset // step
-        self.end_offset = end_offset // step
+        self.start_offset = start_offset
+        self.end_offset = end_offset
 
         # Keep everything **on CPU**
         self.X_feat = data_dict["X_feat"]