diff --git a/Intune.USB.Creator/Public/Publish-ImageToUSB.ps1 b/Intune.USB.Creator/Public/Publish-ImageToUSB.ps1
index 30cd44f..bdab6e5 100644
--- a/Intune.USB.Creator/Public/Publish-ImageToUSB.ps1
+++ b/Intune.USB.Creator/Public/Publish-ImageToUSB.ps1
@@ -7,6 +7,9 @@ function Publish-ImageToUSB {
         [parameter(Mandatory = $true)]
         [string]$windowsIsoPath,
 
+        [parameter(Mandatory = $false)]
+        [int32]$imageIndex,
+
         [parameter(Mandatory = $false)]
         [switch]$getAutoPilotCfg
     )
@@ -42,8 +45,15 @@ function Publish-ImageToUSB {
         }
         #endregion
         #region get image index from wim
-        Write-Host "`nGetting image index from install.wim.." -ForegroundColor Yellow
-        Get-ImageIndexFromWim -wimPath $usb.WIMFilePath -destination "$($usb.downloadPath)\$($usb.dirName2)"
+        if ($imageIndex) {
+            @{
+                "ImageIndex" = $imageIndex
+            } | ConvertTo-Json | Out-File "$($usb.downloadPath)\$($usb.dirName2)\imageIndex.json"
+        }
+        else {
+            Write-Host "`nGetting image index from install.wim.." -ForegroundColor Yellow
+            Get-ImageIndexFromWim -wimPath $usb.WIMFilePath -destination "$($usb.downloadPath)\$($usb.dirName2)"
+        }
         #endregion
         #region get Autopilot config from azure
         if ($getAutopilotCfg) {