Diplomacy screen bugfix