diff --git a/packages/builder-vite/codegen-iframe-script.ts b/packages/builder-vite/codegen-iframe-script.ts
index a4f6ddc9..cc6718f9 100644
--- a/packages/builder-vite/codegen-iframe-script.ts
+++ b/packages/builder-vite/codegen-iframe-script.ts
@@ -30,7 +30,8 @@ export async function generateIframeScriptCode(
       addParameters,
       addLoader,
       addArgTypesEnhancer,
-      addArgsEnhancer
+      addArgsEnhancer,
+      setGlobalRender
     } from '@storybook/client-api';
     import { logger } from '@storybook/client-logger';
     ${absoluteFilesToImport(configEntries, 'config')}
@@ -63,6 +64,9 @@ export async function generateIframeScriptCode(
           case 'argsEnhancers': {
             return value.forEach((enhancer) => addArgsEnhancer(enhancer))
           }
+          case 'render': {
+            return setGlobalRender(value)
+          }
           case 'globals':
           case 'globalTypes': {
             const v = {};