From 254578982bdcceb58eb04b763420f6afc5bd35b7 Mon Sep 17 00:00:00 2001
From: asimon-1 <40246417+asimon-1@users.noreply.github.com>
Date: Mon, 19 Aug 2024 05:17:15 -0700
Subject: [PATCH] Fix random spells being picked for hero (#694)

---
 src/training/buff.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/training/buff.rs b/src/training/buff.rs
index da7987b..98b70fe 100644
--- a/src/training/buff.rs
+++ b/src/training/buff.rs
@@ -104,7 +104,7 @@ pub unsafe fn handle_buffs(
 }
 
 unsafe fn buff_hero(module_accessor: &mut app::BattleObjectModuleAccessor, status: i32) -> bool {
-    let buff_vec: Vec<BuffOption> = MENU.buff_state.to_vec();
+    let buff_vec: Vec<BuffOption> = MENU.buff_state.hero_buffs().to_vec();
     if !is_buffing(module_accessor) {
         // Initial set up for spells
         start_buff(module_accessor);