diff --git a/Add-Thief-Ball.md b/Add-Thief-Ball.md
index 6684d57..d8e0d89 100644
--- a/Add-Thief-Ball.md
+++ b/Add-Thief-Ball.md
@@ -768,8 +768,9 @@ index b22577455..b8605454e 100644
  static void Cmd_trysetcaughtmondexflags(void)
  {
 -    u16 species = GetMonData(&gEnemyParty[0], MON_DATA_SPECIES, NULL);
+-    u32 personality = GetMonData(&gEnemyParty[0], MON_DATA_PERSONALITY, NULL);
 +    u16 species = GetMonData(&gEnemyParty[gBattlerPartyIndexes[BATTLE_OPPOSITE(gBattlerAttacker)]], MON_DATA_SPECIES, NULL);
-     u32 personality = GetMonData(&gEnemyParty[0], MON_DATA_PERSONALITY, NULL);
++    u32 personality = GetMonData(&gEnemyParty[gBattlerPartyIndexes[BATTLE_OPPOSITE(gBattlerAttacker)]], MON_DATA_PERSONALITY, NULL);
  
      if (GetSetPokedexFlag(SpeciesToNationalPokedexNum(species), FLAG_GET_CAUGHT))
      {