diff --git a/pk3DS/Subforms/RSTE.cs b/pk3DS/Subforms/RSTE.cs
index 4ad1ca0..1e3ad21 100644
--- a/pk3DS/Subforms/RSTE.cs
+++ b/pk3DS/Subforms/RSTE.cs
@@ -537,7 +537,12 @@ private void Randomize()
 
                 // Randomize Trainer Stats
                 if (rDiffAI)
-                    CB_AI.SelectedIndex = CB_AI.Items.Count - 1; // max
+                {
+                    if (CB_Battle_Type.SelectedIndex == 0)
+                        CB_AI.SelectedIndex = 7; // Max Single
+                    else if (CB_Battle_Type.SelectedIndex == 1)
+                        CB_AI.SelectedIndex = 135; // Max Double
+                }
                 if (rClass)
                     CB_Trainer_Class.SelectedIndex = (int)(rnd32() % (CB_Trainer_Class.Items.Count));