Fixed import statements in battle.