use correct step when energies present

Becksteinlab · Oct 21, 2024 · 3d100bb · 3d100bb
1 parent f9c8acb
commit 3d100bb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/imdclient/IMDREADER.py b/imdclient/IMDREADER.py
@@ -101,7 +101,7 @@ def _load_imdframe_into_ts(self, imdf):
             self.ts.data["dt"] = imdf.dt
             self.ts.data["step"] = imdf.step
         if imdf.energies is not None:
-            self.ts.data.update(imdf.energies)
+            self.ts.data.update({k: v for k, v in imdf.energies.items() if k != "step"})
         if imdf.box is not None:
             self.ts.dimensions = core.triclinic_box(*imdf.box)
         if imdf.positions is not None: