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.