1 files changed, 3 insertions, 2 deletions
diff --git a/src/client/views/nodes/generativeFill/GenerativeFill.tsx b/src/client/views/nodes/generativeFill/GenerativeFill.tsx
index 7230aa9b1..9c03600cf 100644
--- a/src/client/views/nodes/generativeFill/GenerativeFill.tsx
+++ b/src/client/views/nodes/generativeFill/GenerativeFill.tsx
@@ -20,6 +20,7 @@ import { OpenWhereMod } from '../DocumentView';
 import Buttons from './GenerativeFillButtons';
 import { List } from '../../../../fields/List';
 import { CgClose } from 'react-icons/cg';
+import { ImageBox } from '../ImageBox';
 
 enum BrushStyle {
     ADD,
@@ -408,8 +409,8 @@ const GenerativeFill = ({ imageEditorOpen, imageEditorSource, imageRootDoc, addD
 
     // Closes the editor view
     const handleViewClose = () => {
-        MainView.Instance.setImageEditorOpen(false);
-        MainView.Instance.setImageEditorSource('');
+        ImageBox.setImageEditorOpen(false);
+        ImageBox.setImageEditorSource('');
         if (newCollectionRef.current) {
             newCollectionRef.current.fitContentOnce = true;
         }