diff --git a/Fix-EmberkomCloudBackupProductEdition.ps1 b/Fix-EmberkomCloudBackupProductEdition.ps1
index 6c9f189..a651de5 100644
--- a/Fix-EmberkomCloudBackupProductEdition.ps1
+++ b/Fix-EmberkomCloudBackupProductEdition.ps1
@@ -3,10 +3,10 @@ If (! $Edition ) { $Edition = 'desktop' }
 
 # Set the correct download URL and agent edition
 switch ( $Edition.ToLower() ) {
-    'server'  { $URL = $ECB_WIN_URL; $AgentEdition = 'baremetal' }
-    'desktop' { $URL = $ECB_WIN_URL; $AgentEdition = 'desktop'   }
-    'vm'      { $URL = $ECB_VMM_URL; $AgentEdition = 'vmedition' }
-    default   { $URL = $ECB_WIN_URL; $AgentEdition = 'desktop'   }
+    'server'  { $AgentEdition = 'baremetal' }
+    'desktop' { $AgentEdition = 'desktop'   }
+    'vm'      { $AgentEdition = 'vmedition' }
+    default   { $AgentEdition = 'desktop'   }
 }
 
 # Install and import additional Powershell module