diff --git a/patch-iso.bat b/patch-iso.bat
index 4c5dac5..31bad5f 100644
--- a/patch-iso.bat
+++ b/patch-iso.bat
@@ -22,6 +22,7 @@ if exist "tmp" (
 	echo.
 	echo -- removing tmp folder --
 	rmdir "tmp" /s /q
+	echo.
 )
 
 if "%1" == "" (
@@ -47,8 +48,11 @@ mkdir tmp\disc-image
 echo.
 echo.
 echo -- mounting wmi --
+dism /get-imageinfo /imagefile:"tmp\disc-image\sources\install.wim"
+set /p "index=index: "
+
 mkdir tmp\wim
-dism /mount-image /index:1 /imagefile:"tmp\disc-image\sources\install.wim" /mountdir:"tmp\wim"
+dism /mount-image /index:%index% /imagefile:"tmp\disc-image\sources\install.wim" /mountdir:"tmp\wim"
 
 :: image console mode
 if "%2" == "console" (