From 0d2d23741e520e610f24f3a4fb004fd94f062ac5 Mon Sep 17 00:00:00 2001
From: mustard <mhqnwt@gmail.com>
Date: Wed, 24 Aug 2022 17:28:45 +0000
Subject: [PATCH] 1

---
 components/supervisor/pkg/supervisor/services.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/components/supervisor/pkg/supervisor/services.go b/components/supervisor/pkg/supervisor/services.go
index c5c4a1b5f4c964..b4fdc273740345 100644
--- a/components/supervisor/pkg/supervisor/services.go
+++ b/components/supervisor/pkg/supervisor/services.go
@@ -662,7 +662,7 @@ func (is *InfoService) WorkspaceInfo(ctx context.Context, req *api.WorkspaceInfo
 		} else {
 			for _, wsCls := range classes {
 				if wsCls.ID == is.cfg.WorkspaceClass {
-					resp.WorkspaceClass = &api.WorkspaceInfoResponse_WorkspaceClass{
+					found := &api.WorkspaceInfoResponse_WorkspaceClass{
 						Id:          wsCls.ID,
 						Category:    wsCls.Category,
 						DisplayName: wsCls.DisplayName,
@@ -670,6 +670,8 @@ func (is *InfoService) WorkspaceInfo(ctx context.Context, req *api.WorkspaceInfo
 						Powerups:    uint32(wsCls.Powerups),
 						IsDefault:   wsCls.IsDefault,
 					}
+					resp.WorkspaceClass = found
+					is.workspaceClass = found
 				}
 			}
 		}