diff --git a/src/mol-plugin/state/transforms/representation.ts b/src/mol-plugin/state/transforms/representation.ts
index 417d200b97ff1c4c929d4f765f2e2fd85285a9ff..5e993e95f558cd92084045e86a5b55b292b6ae79 100644
--- a/src/mol-plugin/state/transforms/representation.ts
+++ b/src/mol-plugin/state/transforms/representation.ts
@@ -157,7 +157,7 @@ const StructureRepresentation3D = PluginStateTransform.BuiltIn({
         return ({
             type: PD.Mapped<any>(
                 registry.default.name,
-                registry.types,
+                registry.getApplicableTypes(a.data),
                 name => PD.Group<any>(registry.get(name).getParams(themeCtx, a.data))),
             colorTheme: PD.Mapped<any>(
                 type.defaultColorTheme,