Fix decorator

atomicojs · Mar 17, 2024 · e5b0f47 · e5b0f47
1 parent 2d7ae8f
commit e5b0f47
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/packages/plugin/package.json b/packages/plugin/package.json
@@ -1,6 +1,6 @@
 {
     "name": "@atomico/storybook",
-    "version": "2.6.2",
+    "version": "2.6.3",
     "public": true,
     "description": "Render Atomico's virtualDOM in Storybook",
     "type": "module",

diff --git a/packages/plugin/src/decorator.tsx b/packages/plugin/src/decorator.tsx
@@ -60,10 +60,7 @@ export const decorator =
     (Story: () => any, context: StoryContext) => {
         const cache = context.canvasElement as HTMLElement;
 
-        useStorybookEffect(() => {
-            const story = Story();
-            render(<host>{story}</host>, cache);
-        }, [context.id]);
+        useStorybookEffect(() => () => host.remove(), []);
 
         if (!cache[context.id]) {
             // Avoid double insertion by @storybook/web-components
@@ -116,5 +113,5 @@ export const decorator =
             return "";
         }
 
-        return new Comment("Atomico.decorator");
+        return host;
     };