diff --git a/src/battle_setup.c b/src/battle_setup.c
index 6c3b8d05572..68c3d90ac97 100644
--- a/src/battle_setup.c
+++ b/src/battle_setup.c
@@ -46,6 +46,7 @@
 #include "constants/trainers.h"
 #include "constants/trainer_hill.h"
 #include "constants/weather.h"
+#include "debug.h"
 
 enum {
     TRANSITION_TYPE_NORMAL,
@@ -1392,6 +1393,9 @@ void BattleSetup_StartTrainerBattle_Debug(void)
 
 static void CB2_EndTrainerBattle(void)
 {
+    #if TX_DEBUG_SYSTEM_ENABLE == TRUE
+        gIsDebugBattle = FALSE;
+    #endif
     if (gTrainerBattleOpponent_A == TRAINER_SECRET_BASE)
     {
         SetMainCallback2(CB2_ReturnToFieldContinueScriptPlayMapMusic);