Adjust logic for restart.