Restart from checkpoints ignores the init_step parameter
If the simulation is restarted from the checkpoint, the `init_step` parameter is ignored when the checks if the trajectory is finished are done.
issue