diff --git a/metastable_baselines/ppo/policies.py b/metastable_baselines/ppo/policies.py
index 231a1bf..94d13d0 100644
--- a/metastable_baselines/ppo/policies.py
+++ b/metastable_baselines/ppo/policies.py
@@ -299,7 +299,7 @@ class ActorCriticPolicy(BasePolicy):
         values = self.value_net(latent_vf)
         distribution = self._get_action_dist_from_latent(latent_pi)
         if self.use_pca:
-            actions = distribution.get_actions(deterministic=deterministic, traj=trajectory)
+            actions = distribution.get_actions(deterministic=deterministic, trajectory=trajectory)
         else:
             actions = distribution.get_actions(deterministic=deterministic)
         log_prob = distribution.log_prob(actions)