From aa5a2359530541eeb730e80cab44bbe434d26549 Mon Sep 17 00:00:00 2001
From: Jan De Dobbeleer <jan.de.dobbeleer@gmail.com>
Date: Fri, 24 Sep 2021 07:30:35 +0200
Subject: [PATCH] fix(pwsh): validate correct AZ context object

---
 src/init/omp.ps1 | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/init/omp.ps1 b/src/init/omp.ps1
index 843ba237..0f227805 100644
--- a/src/init/omp.ps1
+++ b/src/init/omp.ps1
@@ -58,7 +58,7 @@ function global:Initialize-ModuleSupport {
     if ($env:AZ_ENABLED -eq $true) {
         try {
             $context = Get-AzContext
-            if ($null -ne $subscription) {
+            if ($null -ne $context) {
                 $env:AZ_ENVIRONMENT_NAME = $context.Environment.Name
                 $env:AZ_USER_NAME = $context.Account.Id
                 $env:AZ_SUBSCRIPTION_ID = $context.Subscription.Id