Fixed assertion error with initial checkpoints