diff --git a/app/react/src/client/preview/render.tsx b/app/react/src/client/preview/render.tsx
index 94db184a3b6b..e2f305158a58 100644
--- a/app/react/src/client/preview/render.tsx
+++ b/app/react/src/client/preview/render.tsx
@@ -147,4 +147,6 @@ export async function renderToDOM(
}
await renderElement(element, domElement);
+
+ return () => unmountElement(domElement);
}
diff --git a/examples/react-ts/.storybook/main.ts b/examples/react-ts/.storybook/main.ts
index 752f85c6ac60..33fb89ce230f 100644
--- a/examples/react-ts/.storybook/main.ts
+++ b/examples/react-ts/.storybook/main.ts
@@ -37,7 +37,7 @@ const config: StorybookConfig = {
},
features: {
postcss: false,
- // modernInlineRender: true,
+ modernInlineRender: true,
storyStoreV7: !global.navigator?.userAgent?.match?.('jsdom'),
buildStoriesJson: true,
babelModeV7: true,
diff --git a/examples/react-ts/src/__snapshots__/storyshots.test.ts.snap b/examples/react-ts/src/__snapshots__/storyshots.test.ts.snap
index 409f2a8ab57e..a3430706744a 100644
--- a/examples/react-ts/src/__snapshots__/storyshots.test.ts.snap
+++ b/examples/react-ts/src/__snapshots__/storyshots.test.ts.snap
@@ -4041,6 +4041,51 @@ exports[`Storyshots Demo/Examples / Emoji Button With Args 1`] = `
`;
+exports[`Storyshots Demo/button2 One 1`] = `
+
+`;
+
+exports[`Storyshots Demo/button2 Three 1`] = `
+
+`;
+
+exports[`Storyshots Demo/button2 Two 1`] = `
+
+`;
+
+exports[`Storyshots Demo/button3 Five 1`] = `
+
+`;
+
+exports[`Storyshots Demo/button3 Four 1`] = `
+
+`;
+
exports[`Storyshots Docs/ButtonMdx Basic 1`] = `