fixed colorTheme.structure assignment

src/mol-geo/representation/structure/units-visual.ts

@@ -71,6 +71,7 @@ export function UnitsMeshVisual<P extends UnitsMeshProps>(builder: UnitsMeshVisu
         if (!renderObject) return
 
         const newProps = Object.assign({}, currentProps, props)
+        newProps.colorTheme.structure = currentStructure
         const unit = currentGroup.units[0]
 
         locationIt.reset()
@@ -131,7 +132,6 @@ export function UnitsMeshVisual<P extends UnitsMeshProps>(builder: UnitsMeshVisu
             } else {
                 if (group && !areGroupsIdentical(group, currentGroup)) {
                     currentGroup = group
-                    currentProps.colorTheme.structure = currentStructure
                 }
                 await update(ctx, props)
             }