fix simple generation to not have checkpoints on very end