1 files changed, 4 insertions, 4 deletions

diff --git a/packages/preload/src/contextBridge/SophieRendererImpl.ts b/packages/preload/src/contextBridge/SophieRendererImpl.ts
index 18ab07e..4c24b74 100644
--- a/packages/preload/src/contextBridge/SophieRendererImpl.ts
+++ b/packages/preload/src/contextBridge/SophieRendererImpl.ts
@@ -35,7 +35,7 @@ class SophieRendererImpl implements SophieRenderer {
 
   private listener: SharedStoreListener | null = null;
 
-  constructor() {
+  constructor(private readonly allowReplaceListener: boolean) {
     ipcRenderer.on(MainToRendererIpcMessage.SharedStorePatch, (_event, patch) => {
       try {
         // `mobx-state-tree` will validate the patch, so we can safely cast here.
@@ -48,7 +48,7 @@ class SophieRendererImpl implements SophieRenderer {
   }
 
   async onSharedStoreChange(listener: SharedStoreListener): Promise<void> {
-    if (this.onSharedStoreChangeCalled) {
+    if (this.onSharedStoreChangeCalled && !this.allowReplaceListener) {
       throw new Error('Shared store change listener was already set');
     }
     this.onSharedStoreChangeCalled = true;
@@ -86,8 +86,8 @@ class SophieRendererImpl implements SophieRenderer {
   }
 }
 
-export function createSophieRenderer(): SophieRenderer {
-  const impl = new SophieRendererImpl();
+export function createSophieRenderer(allowReplaceListener: boolean): SophieRenderer {
+  const impl = new SophieRendererImpl(allowReplaceListener);
   return {
     onSharedStoreChange: impl.onSharedStoreChange.bind(impl),
     dispatchAction: impl.dispatchAction.bind(impl),