diff --git a/assets/win11x64-iot.cmd b/assets/win11x64-iot.cmd
index 4be51a14..be2508ac 100644
--- a/assets/win11x64-iot.cmd
+++ b/assets/win11x64-iot.cmd
@@ -6,11 +6,16 @@ set "SETUP_STARTED=%SCRIPT_DIR%setup.started"
 set "SETUP_COMPLETE=%SCRIPT_DIR%setup.complete"
 
 if "%~1"=="" goto setup
+if /i "%~1"=="pe" goto pe
 if /i "%~1"=="setup" goto setup
 if /i "%~1"=="logon" goto logon
 if /i "%~1"=="specialize" goto specialize
 exit /b 2
 
+:pe
+
+exit /b 0
+
 :specialize
 
 rem Install the VMWare display driver before Windows Setup's final reboot.