diff --git a/comfyui.bat b/comfyui.bat
index 63dfe18a5..17771eea1 100644
--- a/comfyui.bat
+++ b/comfyui.bat
@@ -9,7 +9,11 @@ set COMMANDLINE_ARGS=--auto-launch --use-quad-cross-attention --reserve-vram 0.9
 set ZLUDA_COMGR_LOG_LEVEL=1
 
 echo *** Checking and updating to new version if possible 
+
+copy comfy\customzluda\zluda-default.py comfy\zluda.py /y >NUL
 git pull
+copy comfy\customzluda\zluda-default.py comfy\zluda.py /y >NUL
+
 echo.
 .\zluda\zluda.exe -- %PYTHON% main.py %COMMANDLINE_ARGS%
 pause