1 files changed, 7 insertions, 4 deletions
diff --git a/src/client/views/nodes/DataVizBox/DataVizBox.tsx b/src/client/views/nodes/DataVizBox/DataVizBox.tsx
index 1243831c8..14e0a8e66 100644
--- a/src/client/views/nodes/DataVizBox/DataVizBox.tsx
+++ b/src/client/views/nodes/DataVizBox/DataVizBox.tsx
@@ -658,7 +658,7 @@ export class DataVizBox extends ViewBoxAnnotatableComponent<FieldViewProps>() {
     };
 
     applyLayout = (templateInfo: DataVizTemplateInfo, docs: Doc[]) => {
-        if (templateInfo.layout.type === LayoutType.Stacked) return;
+        if (templateInfo.layout.type === LayoutType.STACKED) return;
         const columns: number = templateInfo.columns;
         const xGap: number = templateInfo.layout.xMargin;
         const yGap: number = templateInfo.layout.yMargin;
@@ -694,9 +694,8 @@ export class DataVizBox extends ViewBoxAnnotatableComponent<FieldViewProps>() {
     // }
 
     @action
-    createDocsFromTemplate = (templateInfo: DataVizTemplateInfo) => {
-        if (!templateInfo.doc) return;
-        const mainCollection = this.DocumentView?.().containerViewPath?.().lastElement()?.ComponentView as CollectionFreeFormView;
+    createDocsFromTemplate = (templateInfo: DataVizTemplateInfo, returnList?: boolean): Doc[] => {
+        if (!templateInfo.doc) {console.log('noinfo'); return []; }
         const fields: string[] = Array.from(Object.keys(this.records[0]));
         const selectedRows = NumListCast(this.layoutDoc.dataViz_selectedRows);
         const docs: Doc[] = selectedRows.map(row => {
@@ -718,9 +717,13 @@ export class DataVizBox extends ViewBoxAnnotatableComponent<FieldViewProps>() {
             return applied;
         });
 
+        if (returnList) return docs;
+
+        const mainCollection = this.DocumentView?.().containerViewPath?.().lastElement()?.ComponentView as CollectionFreeFormView;
         docs.forEach(doc => mainCollection.addDocument(doc));
 
         this.applyLayout(templateInfo, docs);
+        return [];
     };
 
     /**