diff --git a/pyproject.toml b/pyproject.toml
index 6bf2d7f..949df04 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -18,7 +18,7 @@ dynamic = ["version"]
 
 dependencies = [
     "lephare",
-    "pz-rail-base",
+    "pz-rail-base>=1.0.3",
     "qp-prob[full]",
 ]
 
diff --git a/src/rail/estimation/algos/lephare.py b/src/rail/estimation/algos/lephare.py
index ceb838a..8e3b812 100644
--- a/src/rail/estimation/algos/lephare.py
+++ b/src/rail/estimation/algos/lephare.py
@@ -81,9 +81,9 @@ class LephareInformer(CatInformer):
         ),
     )
 
-    def __init__(self, args, comm=None):
+    def __init__(self, args, **kwargs):
         """Init function, init config stuff (COPIED from rail_bpz)"""
-        CatInformer.__init__(self, args, comm=comm)
+        super().__init__(args, **kwargs)
         self.lephare_config = self.config["lephare_config"]
         # We need to ensure the requested redshift grid is propagated
         self.zmin=self.config["zmin"]
@@ -204,8 +204,8 @@ class LephareEstimator(CatEstimator):
         ),
     )
 
-    def __init__(self, args, comm=None):
-        CatEstimator.__init__(self, args, comm=comm)
+    def __init__(self, args, **kwargs):
+        super().__init__(args, **kwargs)
         CatEstimator.open_model(self, **self.config)
         self.lephare_config = self.model["lephare_config"]
         Z_STEP=self.model["lephare_config"]["Z_STEP"]