diff --git a/Tools/CreateParentDisk.ps1 b/Tools/CreateParentDisk.ps1
index e3880ef5..f5c4bd2a 100644
--- a/Tools/CreateParentDisk.ps1
+++ b/Tools/CreateParentDisk.ps1
@@ -153,7 +153,7 @@ If (-not $isAdmin) {
             }
 
             #Generate vhdx name
-            if ($edition -eq "Azure Stack HCI"){
+            if ($edition -like "*Azure Stack HCI"){
                 $tempvhdname = switch ($BuildNumber){
                     17784 {
                         "AzSHCI20H2_G2.vhdx"