diff --git a/src/battle_ai_switch_items.c b/src/battle_ai_switch_items.c
index 4924cc5116..45928df3ce 100644
--- a/src/battle_ai_switch_items.c
+++ b/src/battle_ai_switch_items.c
@@ -1001,7 +1001,7 @@ static bool32 AI_ShouldHeal(u8 healAmount)
                 }
 
                 // AI_THINKING_STRUCT->movesetIndex is the array index of the AI's chosen move
-                if (CanIndexMoveFaintTarget(gActiveBattler, i, AI_THINKING_STRUCT->movesetIndex, 0) && WillAIStrikeFirst())
+                if (CanIndexMoveFaintTarget(gActiveBattler, i, AI_THINKING_STRUCT->movesetIndex, 0) && AI_WhoStrikesFirst(gActiveBattler, i) == AI_IS_FASTER)
                 {
                     // We can faint the target and move first -> don't heal
                     shouldHeal = FALSE;