diff --git a/priorConditionedAnnealing/pca.py b/priorConditionedAnnealing/pca.py
index e37e231..b71efaa 100644
--- a/priorConditionedAnnealing/pca.py
+++ b/priorConditionedAnnealing/pca.py
@@ -143,7 +143,7 @@ class PCA_Distribution(SB3_Distribution):
         """
         if deterministic:
             return self.mode()
-        return self.sample(trajectory=trajectory)
+        return self.sample(traj=trajectory)
 
     def sample(self, traj: th.Tensor, f_sigma: int = 1, epsilon=None) -> th.Tensor:
         pi_mean, pi_std = self.distribution.mean, self.distribution.scale