Fix bug in `Battle` protoype refactor