diff --git a/staging/cse/windows/debug/collect-windows-logs.ps1 b/staging/cse/windows/debug/collect-windows-logs.ps1
index e0fc65257ba..0390d001d1c 100644
--- a/staging/cse/windows/debug/collect-windows-logs.ps1
+++ b/staging/cse/windows/debug/collect-windows-logs.ps1
@@ -1,6 +1,10 @@
 # NOTE: Please also update staging/cse/windows/provisioningscripts/loggenerator.ps1 when collecting new logs.
 $ProgressPreference = "SilentlyContinue"
 
+param (
+    [string[]]$Params
+)
+
 function CollectLogsFromDirectory {
   Param(
       [Parameter(Mandatory=$true)]
@@ -52,6 +56,10 @@ $outputLogFile = "$ENV:TEMP\collect-windows-logs-output.log"
 Start-Transcript -Path $outputLogFile
 $paths += $outputLogFile
 
+if ($args.Count -gt 0) {
+  Write-Host "Script's args: $args"
+}
+
 Write-Host "Collecting logs for various Kubernetes components"
 get-childitem c:\k\*.log* -Exclude $lockedFiles | Foreach-Object {
   $paths += $_
@@ -217,14 +225,16 @@ else {
   Write-Host "ctr.exe command not available"
 }
 
-# collect disk usage information
-if (Test-Path "C:\aks-tools\DU\du.exe") {
-  C:\aks-tools\DU\du.exe /accepteula
-  C:\aks-tools\DU\du.exe -l 1 C:\ProgramData\containerd\root\io.containerd.snapshotter.v1.windows\snapshots\ > "$ENV:TEMP\$timeStamp-du-snapshot-folder-size.txt"
-  $paths += "$ENV:TEMP\$timeStamp-du-snapshot-folder-size.txt"
+if ($args.Count -gt 0 -and $args -contains "disk-usage") {
+  # collect disk usage information
+  if (Test-Path "C:\aks-tools\DU\du.exe") {
+    C:\aks-tools\DU\du.exe /accepteula
+    C:\aks-tools\DU\du.exe -l 1 C:\ProgramData\containerd\root\io.containerd.snapshotter.v1.windows\snapshots\ > "$ENV:TEMP\$timeStamp-du-snapshot-folder-size.txt"
+    $paths += "$ENV:TEMP\$timeStamp-du-snapshot-folder-size.txt"
+  }
+  Copy-Item 'C:\ProgramData\containerd\root\io.containerd.snapshotter.v1.windows\metadata.db' "$ENV:TEMP\$timeStamp-snpashot-metadata.db"
+  $paths += "$ENV:TEMP\$timeStamp-snpashot-metadata.db"
 }
-Copy-Item 'C:\ProgramData\containerd\root\io.containerd.snapshotter.v1.windows\metadata.db' "$ENV:TEMP\$timeStamp-snpashot-metadata.db"
-$paths += "$ENV:TEMP\$timeStamp-snpashot-metadata.db"
 
 # log containers, pods and images the CRI plugin is aware of, and their state.
 $res = Get-Command crictl.exe -ErrorAction SilentlyContinue