diff --git a/engine/battle/ai/move.asm b/engine/battle/ai/move.asm
index 94dbd41c5..bfa3d96aa 100644
--- a/engine/battle/ai/move.asm
+++ b/engine/battle/ai/move.asm
@@ -65,7 +65,7 @@ AIChooseMove:
 	ld hl, TrainerClassAttributes + TRNATTR_AI_MOVE_WEIGHTS
 	ld a, [wTrainerClass]
 	dec a
-	ld bc, 7 ; Trainer2AI - Trainer1AI
+	ld bc, NUM_TRAINER_ATTRIBUTES
 	call AddNTimes
 	lb bc, CHECK_FLAG, 0
 	push bc