diff --git a/samples/manage/azure-sql-db-managed-instance/attach-vpn-gateway/attachVPNGatewayAz.ps1 b/samples/manage/azure-sql-db-managed-instance/attach-vpn-gateway/attachVPNGatewayAz.ps1
index ba2fbdd04d..6cfd2ab486 100644
--- a/samples/manage/azure-sql-db-managed-instance/attach-vpn-gateway/attachVPNGatewayAz.ps1
+++ b/samples/manage/azure-sql-db-managed-instance/attach-vpn-gateway/attachVPNGatewayAz.ps1
@@ -24,7 +24,7 @@ function VerifyPSVersion
 
 function Ensure-Login () 
 {
-    $context = Get-AzureRmContext
+    $context = Get-AzContext
     If($context.Subscription -eq $null)
     {
         Write-Host "Loging in ..."