diff --git a/bioptim/optimization/optimal_control_program.py b/bioptim/optimization/optimal_control_program.py
index f0056af71..7c1440297 100644
--- a/bioptim/optimization/optimal_control_program.py
+++ b/bioptim/optimization/optimal_control_program.py
@@ -662,8 +662,7 @@ def _finalize_penalties(
         # Define continuity constraints
         # Prepare phase transitions (Reminder, it is important that parameters are declared before,
         # otherwise they will erase the phase_transitions)
-        self.phase_transitions = PhaseTransitionBuilder(self).prepare_phase_transitions(phase_transitions)
-        # self.phase_transitions = phase_transitions.prepare_phase_transitions(self)
+        self.phase_transitions = PhaseTransitionBuilder(ocp=self).prepare_phase_transitions(phase_transitions)
 
         # Skipping creates an OCP without built-in continuity constraints, make sure you declared constraints elsewhere
         self._declare_continuity()