diff --git a/src/components/image-editor/ImageCanvasGenerationComposerView.tsx b/src/components/image-editor/ImageCanvasGenerationComposerView.tsx
index 49e94694e..db018c487 100644
--- a/src/components/image-editor/ImageCanvasGenerationComposerView.tsx
+++ b/src/components/image-editor/ImageCanvasGenerationComposerView.tsx
@@ -1703,6 +1703,8 @@ export function ImageCanvasGenerationComposerView({
           style={generationComposerStyle}
           setGenerateDialog={setGenerateDialog}
           generationReferenceButtonRef={generationReferenceButtonRef}
+          renderEditorPortal={renderEditorPortal}
+          buildPortalMenuStyle={buildPortalMenuStyle}
           isGenerationReferenceMenuOpen={isGenerationReferenceMenuOpen}
           setIsGenerationReferenceMenuOpen={setIsGenerationReferenceMenuOpen}
           setIsPickingGenerationReferenceFromCanvas={