Fix typo's in the cleaner script

FAForever · Apr 30, 2024 · f81c6e8 · f81c6e8
1 parent 561107f
commit f81c6e8
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tools/cleaner.py b/tools/cleaner.py
@@ -421,16 +421,16 @@ def find_classification(unit):
             return 'RULEUC_Engineer'
         if "COMMAND" in unit['Categories']:
             return 'RULEUC_Commander'
-        if "COUNTERINTELLIGENCE" in unit['Categories']:
-            return 'RULEUC_CounterMeasure'
         if "LAND" in unit['Categories']: 
             return 'RULEUC_MilitaryVehicle'
         if "NAVAL" in unit['Categories']:
             return 'RULEUC_MilitaryShip'
         if "SUB" in unit['Categories']:
             return 'RULEUC_MilitarySub'
         if "AIR" in unit['Categories']:
-            return 'RULEUC_MilitarySub'
+            return 'RULEUC_MilitaryAircraft'
+        if "COUNTERINTELLIGENCE" in unit['Categories']:
+            return 'RULEUC_CounterMeasure'
 
     if "STRUCTURE" in unit['Categories']:
         if "ENGINEER" in unit['Categories']:
@@ -445,6 +445,8 @@ def find_classification(unit):
             return 'RULEUC_MilitaryStructure'
         if "ANTIMISSILE" in unit['Categories']:
             return 'RULEUC_MilitaryStructure'
+        if "COUNTERINTELLIGENCE" in unit['Categories']:
+            return 'RULEUC_CounterMeasure'
         if "SHIELD" in unit['Categories']:
             return 'RULEUC_MiscSupport'
         if "RADAR" in unit['Categories']: