cleanup

2 files changed, 3 insertions, 4 deletions

diff --git a/src/client/views/collections/CollectionDockingView.tsx b/src/client/views/collections/CollectionDockingView.tsx
index a723dc69d..73179a266 100644
--- a/src/client/views/collections/CollectionDockingView.tsx
+++ b/src/client/views/collections/CollectionDockingView.tsx
@@ -129,7 +129,6 @@ export class CollectionDockingView extends CollectionSubView() {
     }
 
     @undoBatch
-    @action
     public static ReplaceTab(document: Doc, mods: OpenWhereMod, stack: any, panelName: string, addToSplit?: boolean, keyValue?: boolean): boolean {
         const instance = CollectionDockingView.Instance;
         if (!instance) return false;
diff --git a/src/client/views/nodes/generativeFill/GenerativeFill.tsx b/src/client/views/nodes/generativeFill/GenerativeFill.tsx
index 91b0ebd5c..25289967e 100644
--- a/src/client/views/nodes/generativeFill/GenerativeFill.tsx
+++ b/src/client/views/nodes/generativeFill/GenerativeFill.tsx
@@ -56,7 +56,7 @@ const GenerativeFill = ({ imageEditorOpen, imageEditorSource, imageRootDoc, addD
     const [isBrushing, setIsBrushing] = useState(false);
     const [canvasScale, setCanvasScale] = useState(0.5);
     // format: array of [image source, corresponding image Doc]
-    const [edits, setEdits] = useState<(string | Doc)[][]>([]);
+    const [edits, setEdits] = useState<{ url: string; saveRes: Doc | undefined }[]>([]);
     const [edited, setEdited] = useState(false);
     // const [brushStyle] = useState<BrushStyle>(BrushStyle.ADD);
     const [input, setInput] = useState('');
@@ -317,7 +317,7 @@ const GenerativeFill = ({ imageEditorOpen, imageEditorSource, imageRootDoc, addD
                     imgUrls.map(async url => {
                         // eslint-disable-next-line no-use-before-define
                         const saveRes = await onSave(url);
-                        return [url, saveRes as Doc];
+                        return { url, saveRes };
                     })
                 );
                 setEdits(imgRes);
@@ -326,7 +326,7 @@ const GenerativeFill = ({ imageEditorOpen, imageEditorSource, imageRootDoc, addD
                 image.src = imgUrls[0];
                 ImageUtility.drawImgToCanvas(image, canvasRef, canvasDims.width, canvasDims.height);
                 currImg.current = image;
-                parentDoc.current = imgRes[0][1] as Doc;
+                parentDoc.current = imgRes[0].saveRes ?? null;
             }
         } catch (err) {
             console.log(err);