diff --git a/sccoral/module/_module.py b/sccoral/module/_module.py
index 0917415..cbc0fa1 100644
--- a/sccoral/module/_module.py
+++ b/sccoral/module/_module.py
@@ -200,7 +200,9 @@ def __init__(
                     n_levels = 1
 
                 name = f"encoder_{cat_name}"
-                model = LinearEncoder(n_levels, 1, distribution=latent_distribution, mean_bias=True, var_bias=True)
+                model = LinearEncoder(
+                    n_levels, 1, latent_distribution=latent_distribution, mean_bias=True, var_bias=True
+                )
 
                 # Register encoder in class
                 setattr(self, name, model)