diff --git a/priorConditionedAnnealing/pca.py b/priorConditionedAnnealing/pca.py
index 1454e53..e37e231 100644
--- a/priorConditionedAnnealing/pca.py
+++ b/priorConditionedAnnealing/pca.py
@@ -118,8 +118,7 @@ class PCA_Distribution(SB3_Distribution):
 
     def proba_distribution_net(self, latent_dim: int):
         mu_net = nn.Linear(latent_dim, self.action_dim)
-        std_net = StdNet(latent_dim, self.action_dim,
-                         self.init_std, self.par_strength)
+        std_net = StdNet(latent_dim, self.action_dim, self.init_std, self.par_strength, self.epsilon)
 
         return mu_net, std_net