From 539e71ad47ecbbd2cbb484320a25eb8433ddbc5c Mon Sep 17 00:00:00 2001
From: sawka <mike.sawka@gmail.com>
Date: Tue, 4 Oct 2022 11:15:59 -0700
Subject: [PATCH] only show remote upgrade if initpk was received

---
 pkg/remote/remote.go | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/pkg/remote/remote.go b/pkg/remote/remote.go
index 225f536e6..10b2fa62d 100644
--- a/pkg/remote/remote.go
+++ b/pkg/remote/remote.go
@@ -897,9 +897,10 @@ func (msh *MShellProc) Launch() {
 		if cproc != nil && cproc.InitPk != nil {
 			msh.Remote.InitPk = cproc.InitPk
 			mshellVersion = cproc.InitPk.Version
-		}
-		if semver.Compare(mshellVersion, MShellVersion) < 0 {
-			msh.NeedsMShellUpgrade = true
+			if semver.Compare(mshellVersion, MShellVersion) < 0 {
+				// only set NeedsMShellUpgrade if we got an InitPk
+				msh.NeedsMShellUpgrade = true
+			}
 		}
 		// no notify here, because we'll call notify in either case below
 	})