diff --git a/components/usage/pkg/db/dbtest/workspace_instance.go b/components/usage/pkg/db/dbtest/workspace_instance.go
index 8c788db36875a3..5ddc244c8509ed 100644
--- a/components/usage/pkg/db/dbtest/workspace_instance.go
+++ b/components/usage/pkg/db/dbtest/workspace_instance.go
@@ -66,10 +66,16 @@ func NewWorkspaceInstance(t *testing.T, instance db.WorkspaceInstance) db.Worksp
 		attributionID = instance.UsageAttributionID
 	}
 
+	workspaceClass := db.WorkspaceClass_Default
+	if instance.WorkspaceClass != "" {
+		workspaceClass = instance.WorkspaceClass
+	}
+
 	return db.WorkspaceInstance{
 		ID:                 id,
 		WorkspaceID:        workspaceID,
 		UsageAttributionID: attributionID,
+		WorkspaceClass:     workspaceClass,
 		Configuration:      nil,
 		Region:             "",
 		ImageBuildInfo:     sql.NullString{},
diff --git a/components/usage/pkg/db/workspace_instance.go b/components/usage/pkg/db/workspace_instance.go
index f730bca2217cfe..2ffc7719989c15 100644
--- a/components/usage/pkg/db/workspace_instance.go
+++ b/components/usage/pkg/db/workspace_instance.go
@@ -25,6 +25,7 @@ type WorkspaceInstance struct {
 	WorkspaceBaseImage string         `gorm:"column:workspaceBaseImage;type:varchar;size:255;" json:"workspaceBaseImage"`
 	WorkspaceImage     string         `gorm:"column:workspaceImage;type:varchar;size:255;" json:"workspaceImage"`
 	UsageAttributionID AttributionID  `gorm:"column:usageAttributionId;type:varchar;size:60;" json:"usageAttributionId"`
+	WorkspaceClass     string         `gorm:"column:workspaceClass;type:varchar;size:255;" json:"workspaceClass"`
 
 	CreationTime VarcharTime `gorm:"column:creationTime;type:varchar;size:255;" json:"creationTime"`
 	StartedTime  VarcharTime `gorm:"column:startedTime;type:varchar;size:255;" json:"startedTime"`
@@ -120,3 +121,7 @@ func (a AttributionID) Values() (entity string, identifier string) {
 
 	return tokens[0], tokens[1]
 }
+
+const (
+	WorkspaceClass_Default = "default"
+)