From ebd3fcba87233d18abcb55d81b705c41ef73b988 Mon Sep 17 00:00:00 2001
From: I531348 <guoda.puidokaite@sap.com>
Date: Fri, 8 Nov 2024 12:36:40 +0100
Subject: [PATCH] chore(ui): slight adjustments to Story

---
 .../src/components/Grid/Grid.stories.tsx            | 10 ++++------
 .../src/components/GridRow/GridRow.stories.tsx      | 13 +++++--------
 2 files changed, 9 insertions(+), 14 deletions(-)

diff --git a/packages/ui-components/src/components/Grid/Grid.stories.tsx b/packages/ui-components/src/components/Grid/Grid.stories.tsx
index 842fba51e..7bbd60be9 100644
--- a/packages/ui-components/src/components/Grid/Grid.stories.tsx
+++ b/packages/ui-components/src/components/Grid/Grid.stories.tsx
@@ -19,18 +19,16 @@ export default {
     },
   },
   decorators: [
-    (Story, context) => (
+    (Story) => (
       <div className="juno-container jn-px-6 jn-py-6">
-        <Story {...context} />
+        <Story />
       </div>
     ),
   ],
 } as Meta<typeof Grid>
 
-const Template: StoryFn<GridProps> = (args, context) => (
-  <Grid {...args} className={`jn-bg-juno-blue-3 jn-text-juno-grey-blue ${args.className}`}>
-    {context.args.children}
-  </Grid>
+const Template: StoryFn<GridProps> = (args) => (
+  <Grid {...args} className={`jn-bg-juno-blue-3 jn-text-juno-grey-blue ${args.className || ""}`} />
 )
 
 export const Default = {
diff --git a/packages/ui-components/src/components/GridRow/GridRow.stories.tsx b/packages/ui-components/src/components/GridRow/GridRow.stories.tsx
index 14cf26e99..baaa4444b 100644
--- a/packages/ui-components/src/components/GridRow/GridRow.stories.tsx
+++ b/packages/ui-components/src/components/GridRow/GridRow.stories.tsx
@@ -18,14 +18,11 @@ export default {
     },
   },
   decorators: [
-    (Story, context) => {
-      const { parameters } = context
-      return (
-        <div className={`jn-bg-juno-blue-3 jn-text-juno-grey-blue ${parameters.className || ""}`}>
-          <Story {...parameters} />
-        </div>
-      )
-    },
+    (Story) => (
+      <div className="jn-bg-juno-blue-3 jn-text-juno-grey-blue">
+        <Story />
+      </div>
+    ),
   ],
 } as Meta<typeof GridRow>