fix RIVAL1 trainer id labeling

original-commit-id: fafb53ebf94e1501bf285c0229fbd61cc3755f35

1 files changed, 5 insertions, 1 deletions

diff --git a/crystal.py b/crystal.py
index 2de5e13..88a7f14 100644
--- a/crystal.py
+++ b/crystal.py
@@ -3626,7 +3626,11 @@ class TrainerHeader:
             seed = self.name
 
         if "?" in seed:
-            seed = trainer_group_names[self.trainer_group_id]["name"]+"_"+seed[-1]
+            if seed[-2].isdigit():
+                x = 2
+            else:
+                x = 1
+            seed = trainer_group_names[self.trainer_group_id]["name"]+"_"+seed[-x:]
         elif self.trainer_group_id == 0x1f and "EXECUTIVE" in seed:
             seed = "GRUNT_"+seed