diff --git a/apax/md/io.py b/apax/md/io.py
index d5579d11..0a312106 100644
--- a/apax/md/io.py
+++ b/apax/md/io.py
@@ -70,7 +70,7 @@ def __init__(
         self.sampling_rate = sampling_rate
         self.traj_path = traj_path
         self.time_step = time_step
-        self.db = znh5md.IO(self.traj_path, timestep=self.time_step)
+        self.db = znh5md.IO(self.traj_path, timestep=self.time_step, store="time")
 
         self.step_counter = 0
         self.buffer = []