diff --git a/lib/login.js b/lib/login.js
index 7cd18d0f..07d53d84 100644
--- a/lib/login.js
+++ b/lib/login.js
@@ -91,10 +91,12 @@ function getKubectlPath() {
         return kubectlPath;
     });
 }
-function setContext() {
+function setContext(kubeconfigPath) {
     return __awaiter(this, void 0, void 0, function* () {
         let context = core.getInput('context');
         if (context) {
+            //To use kubectl commands, the environment variable KUBECONFIG needs to be set for this step 
+            process.env['KUBECONFIG'] = kubeconfigPath;
             const kubectlPath = yield getKubectlPath();
             let toolRunner = new toolrunner_1.ToolRunner(kubectlPath, ['config', 'use-context', context]);
             yield toolRunner.exec();
@@ -112,7 +114,7 @@ function run() {
         fs.writeFileSync(kubeconfigPath, kubeconfig);
         command_1.issueCommand('set-env', { name: 'KUBECONFIG' }, kubeconfigPath);
         console.log('KUBECONFIG environment variable is set');
-        yield setContext();
+        yield setContext(kubeconfigPath);
     });
 }
 exports.run = run;
diff --git a/src/login.ts b/src/login.ts
index 5e2837ea..57c38a83 100644
--- a/src/login.ts
+++ b/src/login.ts
@@ -89,9 +89,11 @@ async function getKubectlPath() {
     return kubectlPath;
 }
 
-async function setContext() {
+async function setContext(kubeconfigPath: string) {
     let context = core.getInput('context');
     if (context) {
+        //To use kubectl commands, the environment variable KUBECONFIG needs to be set for this step 
+        process.env['KUBECONFIG'] = kubeconfigPath;
         const kubectlPath = await getKubectlPath();
         let toolRunner = new ToolRunner(kubectlPath, ['config', 'use-context', context]);
         await toolRunner.exec();
@@ -108,7 +110,7 @@ export async function run() {
     fs.writeFileSync(kubeconfigPath, kubeconfig);
     issueCommand('set-env', { name: 'KUBECONFIG' }, kubeconfigPath);
     console.log('KUBECONFIG environment variable is set');
-    await setContext();
+    await setContext(kubeconfigPath);
 }
 
 run().catch(core.setFailed);
\ No newline at end of file