diff --git a/packages/react/src/components/EmbeddedIFrameWrapper.tsx b/packages/react/src/components/EmbeddedIFrameWrapper.tsx
index c94f4be0..02f16c29 100644
--- a/packages/react/src/components/EmbeddedIFrameWrapper.tsx
+++ b/packages/react/src/components/EmbeddedIFrameWrapper.tsx
@@ -18,7 +18,7 @@ export const EmbeddedIFrameWrapper = (
     handleCloseInstance: () => void
   }
 ): JSX.Element => {
-  const { open, sessionSpace, ready, iframe } = useContext(FlatfileContext)
+  const { open, sessionSpace, ready, iframe, onClose } = useContext(FlatfileContext)
 
   const [showExitWarnModal, setShowExitWarnModal] = useState(false)
   const {
@@ -99,6 +99,7 @@ export const EmbeddedIFrameWrapper = (
             setShowExitWarnModal(false)
             if (closeSpace && typeof closeSpace.onClose === 'function') {
               closeSpace.onClose({})
+              onClose.current?.()
             }
           }}
           onCancel={() => setShowExitWarnModal(false)}