diff --git a/src/DiskPool/custom/New-AzDiskPool.ps1 b/src/DiskPool/custom/New-AzDiskPool.ps1
index 5b60234484fd..3775529830e0 100644
--- a/src/DiskPool/custom/New-AzDiskPool.ps1
+++ b/src/DiskPool/custom/New-AzDiskPool.ps1
@@ -181,7 +181,7 @@ param(
             if ($PSBoundParameters.ContainsKey("DiskId")){
                 $disk = @()
                 for ($i = 0; $i -lt $DiskId.Count; $i++) {
-                    $diskObject = New-Object -TypeName Microsoft.Azure.PowerShell.Cmdlets.DiskPool.Models.Api20210401Preview.Disk
+                    $diskObject = New-Object -TypeName Microsoft.Azure.PowerShell.Cmdlets.DiskPool.Models.Api20210401Preview.Disk
                     $diskObject.Id = $DiskId[$i]
                     $disk += $diskObject
                 }
diff --git a/src/DiskPool/custom/Update-AzDiskPool.ps1 b/src/DiskPool/custom/Update-AzDiskPool.ps1
index 06e9311b2775..1a0a2f412252 100644
--- a/src/DiskPool/custom/Update-AzDiskPool.ps1
+++ b/src/DiskPool/custom/Update-AzDiskPool.ps1
@@ -159,7 +159,7 @@ param(
             if ($PSBoundParameters.ContainsKey("DiskId")){
                 $disk = @()
                 for ($i = 0; $i -lt $DiskId.Count; $i++) {
-                    $diskObject = New-Object -TypeName Microsoft.Azure.PowerShell.Cmdlets.DiskPool.Models.Api20210401Preview.Disk
+                    $diskObject = New-Object -TypeName Microsoft.Azure.PowerShell.Cmdlets.DiskPool.Models.Api20210401Preview.Disk
                     $diskObject.Id = $DiskId[$i]
                     $disk += $diskObject
                 }