diff --git a/src/battle_main.c b/src/battle_main.c
index cc5f281d4e..fec093d166 100644
--- a/src/battle_main.c
+++ b/src/battle_main.c
@@ -583,6 +583,12 @@ static void CB2_InitBattleInternal(void)
         TryFormChange(i, B_SIDE_OPPONENT, FORM_CHANGE_BEGIN_BATTLE);
     }
 
+    if (TESTING)
+    {
+        gPlayerPartyCount = CalculatePartyCount(gPlayerParty);
+        gEnemyPartyCount = CalculatePartyCount(gEnemyParty);
+    }
+
     gBattleCommunication[MULTIUSE_STATE] = 0;
 }