diff --git a/apis/project.cattle.io/v3/schema/schema.go b/apis/project.cattle.io/v3/schema/schema.go
index 5e87e5f6..8ec8b5ba 100644
--- a/apis/project.cattle.io/v3/schema/schema.go
+++ b/apis/project.cattle.io/v3/schema/schema.go
@@ -154,7 +154,9 @@ func statefulSetTypes(schemas *types.Schemas) *types.Schemas {
 			NewWorkloadTypeMapper(),
 		).
 		MustImport(&Version, v1beta2.StatefulSetSpec{}, statefulSetConfigOverride{}).
-		MustImport(&Version, v1beta2.StatefulSet{}, projectOverride{})
+		MustImportAndCustomize(&Version, v1beta2.StatefulSet{}, func(schema *types.Schema) {
+			schema.BaseType = "workload"
+		}, projectOverride{})
 }
 
 func replicaSetTypes(schemas *types.Schemas) *types.Schemas {