diff --git a/tools/modules/AksEdgeDeploy/AksEdgeDeploy.psm1 b/tools/modules/AksEdgeDeploy/AksEdgeDeploy.psm1
index 7641590..e15c133 100644
--- a/tools/modules/AksEdgeDeploy/AksEdgeDeploy.psm1
+++ b/tools/modules/AksEdgeDeploy/AksEdgeDeploy.psm1
@@ -279,8 +279,7 @@ function UpgradeJsonFormat {
     }
     #upgrade from public preview format to GA format
     $edgeCfg = $jsonObj.AksEdgeConfig
-
-    if ($edgeCfg.SchemaVersion -gt "1.4") {
+    if ([version]$edgeCfg.SchemaVersion -gt [version]"1.4") {
         if (($azCfg.Auth.Password) -and ([string]::IsNullOrEmpty($($edgeCfg.Arc.ClientSecret)))) {
             #Copy over the Azure parameters to Arc section
             $edgeCfg | Add-Member -MemberType NoteProperty -Name 'Arc' -Value $arcdata -Force