diff --git a/bioptim/dynamics/integrator.py b/bioptim/dynamics/integrator.py
index a7802ab08..4e2b789cc 100644
--- a/bioptim/dynamics/integrator.py
+++ b/bioptim/dynamics/integrator.py
@@ -30,6 +30,8 @@ class Integrator:
         The parameters variables scaling factor
     fun: Callable
         The dynamic function which provides the derivative of the states
+    implicit_fun: Callable
+        The implicit dynamic function which provides the defects of the dynamics
     control_type: ControlType
         The type of the controls
     step_time: float
diff --git a/bioptim/dynamics/ode_solver.py b/bioptim/dynamics/ode_solver.py
index 4b8f35642..d6c975bfc 100644
--- a/bioptim/dynamics/ode_solver.py
+++ b/bioptim/dynamics/ode_solver.py
@@ -228,6 +228,10 @@ class COLLOCATION(OdeSolverBase):
         ----------
         polynomial_degree: int
             The degree of the implicit RK
+        method : str
+            The method of interpolation ("legendre" or "radau")
+        defects_type: DefectType
+            The type of defect to use (DefectType.EXPLICIT or DefectType.IMPLICIT)
 
         Methods
         -------
@@ -309,6 +313,10 @@ class IRK(COLLOCATION):
         ----------
         polynomial_degree: int
             The degree of the implicit RK
+        method: str
+            The method of interpolation ("legendre" or "radau")
+        defects_type: DefectType
+            The type of defect to use (DefectType.EXPLICIT or DefectType.IMPLICIT)
 
         Methods
         -------