slight modification to the battle restart system