diff --git a/export_debug.bat b/export_debug.bat
index 06eb50e74..e939f5116 100644
--- a/export_debug.bat
+++ b/export_debug.bat
@@ -4,8 +4,8 @@ REM Avoid footgun by explictly navigating to the directory containing the batch
 cd /d "%~dp0"
 
 REM Verify that OneTrainer is our current working directory
-if not exist "scripts\train_ui.py" (
-    echo Error: train_ui.py does not exist, you have done something very wrong. Reclone the repository.
+if not exist "scripts\train_ui_qt.py" (
+    echo Error: train_ui_qt.py does not exist, you have done something very wrong. Reclone the repository.
     goto :end
 )
 
diff --git a/update.bat b/update.bat
index 7d05c4810..2bd4f6821 100644
--- a/update.bat
+++ b/update.bat
@@ -5,8 +5,8 @@ REM Avoid footgun by explictly navigating to the directory containing the batch
 cd /d "%~dp0"
 
 REM Verify that OneTrainer is our current working directory
-if not exist "scripts\train_ui.py" (
-    echo Error: train_ui.py does not exist, you have done something very wrong. Reclone the repository.
+if not exist "scripts\train_ui_qt.py" (
+    echo Error: train_ui_qt.py does not exist, you have done something very wrong. Reclone the repository.
     goto :end
 )