diff --git a/metastable_baselines/misc/rollout_buffer.py b/metastable_baselines/misc/rollout_buffer.py
index 85eed07..b96e628 100644
--- a/metastable_baselines/misc/rollout_buffer.py
+++ b/metastable_baselines/misc/rollout_buffer.py
@@ -278,6 +278,6 @@ class GaussianRolloutCollectorAuxclass():
 
         return True
 
-    def get_past_trajectories(self):
+    def get_past_trajectories(self) -> th.Tensor:
         # TODO: Respect Episode Boundaries
-        return self.rollout_buffer.actions
+        return th.Tensor(self.rollout_buffer.actions)