Bugfix for test_battle5.py. Again. Because I'm sloppy.