diff --git a/Intune.USB.Creator/Private/Get-DiskToUse.ps1 b/Intune.USB.Creator/Private/Get-DiskToUse.ps1
index a3cc837..4c68526 100644
--- a/Intune.USB.Creator/Private/Get-DiskToUse.ps1
+++ b/Intune.USB.Creator/Private/Get-DiskToUse.ps1
@@ -4,7 +4,7 @@ function Get-DiskToUse {
         [parameter(Mandatory = $false)]
         [int32]$diskNum
     )
-    $diskList = Get-Disk | Where-Object { $_.Bustype -notin @('SATA', 'NVMe') }
+    $diskList = Get-Disk | Where-Object -FilterScript {$_.Bustype -Eq "USB"}
     $disks = $diskList | Select-Object Number, @{Name = 'TotalSize(GB)'; Expression = { [double]($_.Size / 1GB).ToString("#.##") } }, @{Name = "Name"; Expression = { $_.FriendlyName } } | Sort-Object -Property Number
     if (!$diskNum) {
         $table = $disks | Format-Table | Out-Host
@@ -20,4 +20,4 @@ function Get-DiskToUse {
         }
     }
     return $diskNum
-}
\ No newline at end of file
+}