diff --git a/datasets/driving_dataset.py b/datasets/driving_dataset.py
index 38b3e07..15e8da5 100644
--- a/datasets/driving_dataset.py
+++ b/datasets/driving_dataset.py
@@ -585,7 +585,7 @@ def split_train_test(self):
         if self.data_cfg.pixel_source.test_image_stride != 0:
             test_timesteps = np.arange(
                 # it makes no sense to have test timesteps before the start timestep
-                self.data_cfg.pixel_source.test_image_stride,
+                max(0,self.data_cfg.pixel_source.test_image_stride-1),
                 self.num_img_timesteps,
                 self.data_cfg.pixel_source.test_image_stride,
             )