c

1 files changed, 7 insertions, 3 deletions

diff --git a/src/client/views/collections/collectionSchema/SchemaRowBox.tsx b/src/client/views/collections/collectionSchema/SchemaRowBox.tsx
index 8e4161413..8ac2c0314 100644
--- a/src/client/views/collections/collectionSchema/SchemaRowBox.tsx
+++ b/src/client/views/collections/collectionSchema/SchemaRowBox.tsx
@@ -74,9 +74,13 @@ export class SchemaRowBox extends ViewBoxBaseComponent<SchemaRowBoxProps>() {
         });
         if (this.Document['type'] === 'collection') {
                 ContextMenu.Instance.addItem({
-                description: 'Add children to schema',
-                event: () => this.schemaView.addDocsFromChildCollection(this.Document as unknown as CollectionSchemaView),
-                icon: 'plus',
+                description: this.Document._childrenSharedWithSchema ? 'Remove children from schema' : 'Add children to schema',
+                event: () => {
+                    this.Document._childrenSharedWithSchema = !this.Document._childrenSharedWithSchema; 
+                    this.Document._childrenSharedWithSchema ? 
+                    this.schemaView.addDocsFromChildCollection(this.Document) : this.schemaView.removeChildCollectionDocs(this.Document);
+                },
+                icon: this.Document._childrenSharedWithSchema ? 'minus' : 'plus',
             });
         }
         ContextMenu.Instance.displayMenu(x, y, undefined, false);