diff --git a/src/training/buff.rs b/src/training/buff.rs
index d5f04e3..da7987b 100644
--- a/src/training/buff.rs
+++ b/src/training/buff.rs
@@ -111,7 +111,7 @@ unsafe fn buff_hero(module_accessor: &mut app::BattleObjectModuleAccessor, statu
         set_buff_rem(module_accessor, buff_vec.len());
         // Since it's the first step of buffing, we need to set up how many buffs there are
     }
-    if get_buff_rem(module_accessor) <= 0 {
+    if get_buff_rem(module_accessor) == 0 {
         // If there are no buffs selected/left, we're done
         if frame_counter::should_delay(3_u32, *BUFF_DELAY_COUNTER) {
             // Need to wait 3 frames to make sure we stop the spell SFX, since it's a bit delayed
diff --git a/src/training/ui/mod.rs b/src/training/ui/mod.rs
index 2a6290f..1cc3055 100644
--- a/src/training/ui/mod.rs
+++ b/src/training/ui/mod.rs
@@ -18,7 +18,7 @@ mod input_log;
 pub mod menu;
 pub mod notifications;
 
-pub unsafe fn set_icon_text(pane: &mut TextBox, icons: &Vec<u16>) {
+pub unsafe fn set_icon_text(pane: &mut TextBox, icons: &[u16]) {
     pane.set_text_string("");
 
     let it = pane.text_buf as *mut u16;