Small clean-up to damage_generator.py.