diff --git a/src/training/input_record.rs b/src/training/input_record.rs
index 7625be6..bd79d54 100644
--- a/src/training/input_record.rs
+++ b/src/training/input_record.rs
@@ -1,15 +1,20 @@
-use crate::common::button_config;
-use crate::common::consts::{FighterId, HitstunPlayback, OnOff, RecordTrigger};
-use crate::common::input::*;
-use crate::common::{get_module_accessor, is_in_hitstun, is_in_shieldstun, MENU};
-use crate::training::mash;
-use crate::training::ui::notifications::{clear_notifications, color_notification};
+use std::cmp::Ordering;
+
 use lazy_static::lazy_static;
 use parking_lot::Mutex;
 use skyline::nn::ui2d::ResColor;
 use smash::app::{lua_bind::*, utility, BattleObjectModuleAccessor};
 use smash::lib::lua_const::*;
-use std::cmp::Ordering;
+
+use InputRecordState::*;
+use PossessionState::*;
+
+use crate::common::consts::{FighterId, HitstunPlayback, OnOff, RecordTrigger};
+use crate::common::input::*;
+use crate::common::{button_config, is_training_mode};
+use crate::common::{get_module_accessor, is_in_hitstun, is_in_shieldstun, MENU};
+use crate::training::mash;
+use crate::training::ui::notifications::{clear_notifications, color_notification};
 
 #[derive(PartialEq, Debug)]
 pub enum InputRecordState {
@@ -43,9 +48,6 @@ pub enum StartingStatus {
     Other,
 }
 
-use InputRecordState::*;
-use PossessionState::*;
-
 const STICK_NEUTRAL: f32 = 0.2;
 const STICK_CLAMP_MULTIPLIER: f32 = 1.0 / 120.0; // 120.0 = CLAMP_MAX
 const FINAL_RECORD_MAX: usize = 600; // Maximum length for input recording sequences (capacity)