Fix crash in Battle prototype refactor