1 files changed, 6 insertions, 2 deletions

diff --git a/cmd/client/client.go b/cmd/client/client.go
index b72003d..afc7a1d 100644
--- a/cmd/client/client.go
+++ b/cmd/client/client.go
@@ -37,7 +37,7 @@ func NewClientCommand(appVersion string) *cobra.Command {
                 Run: func(c *cobra.Command, args []string) {
                         cfg := app.Config{}
                         cli.MustGetConfig(c, &cfg)
-                        clientMain(cfg, args[0], args[1], args[2])
+                        clientMain(cfg, appVersion, args[0], args[1], args[2])
                 },
         }
         cli.AddFlags(clientCmd, &app.Config{}, app.DefaultConfig, "client")
@@ -202,12 +202,16 @@ func fetchOauthToken(ctx context.Context, clientId, proxyHost string) (string, e
         return tokenResponse.AccessToken, nil
 }
 
-func clientMain(cfg app.Config, host, port, username string) {
+func clientMain(cfg app.Config, appVersion, host, port, username string) {
         log.SetOutput(os.Stderr)
 
         ctx, cancel := context.WithCancel(context.Background())
         defer cancel()
 
+        if err := upgradeAndRestartClient(ctx, appVersion, cfg.ClientHost); err != nil {
+                log.Fatalf("Error attempting to upgrade client: %s", err)
+        }
+
         agentConn, err := connectToAgent()
         if err != nil {
                 log.Fatalf("Error connecting to agent, is it started?")