diff --git a/metastable_baselines/misc/rollout_buffer.py b/metastable_baselines/misc/rollout_buffer.py
index 846d60a..1fea969 100644
--- a/metastable_baselines/misc/rollout_buffer.py
+++ b/metastable_baselines/misc/rollout_buffer.py
@@ -236,6 +236,8 @@ class GaussianRolloutCollectorAuxclass():
             if callback.on_step() is False:
                 return False
 
+            if isinstance(env, DummyVecEnv):
+                self._update_info_buffer(infos[0])
             self._update_info_buffer(infos)
             n_steps += 1