diff --git a/metastable_baselines/sac/policies.py b/metastable_baselines/sac/policies.py
index a01d64c..010d5c8 100644
--- a/metastable_baselines/sac/policies.py
+++ b/metastable_baselines/sac/policies.py
@@ -320,7 +320,7 @@ class SACPolicy(BasePolicy):
         }
 
         self.actor_kwargs.update(sde_kwargs)
-        self.actor.kwargs.update({'latent_dim_sde': latent_dim_sde})
+        self.actor_kwargs.update({'latent_dim_sde': latent_dim_sde})
         self.critic_kwargs = self.net_args.copy()
         self.critic_kwargs.update(
             {