diff --git a/packages/ui/src/components/Flex/Flex.stories.tsx b/packages/ui/src/components/Flex/Flex.stories.tsx
index 940c71f272..dac4f530c1 100644
--- a/packages/ui/src/components/Flex/Flex.stories.tsx
+++ b/packages/ui/src/components/Flex/Flex.stories.tsx
@@ -283,9 +283,7 @@ export const FlexItems = meta.story({
       options: [undefined, '0%', '25%', '50%', '100%', 'auto'],
     },
   },
-  render: args => {
-    const Component = args.component;
-
+  render: ({ component: Component, ...args }) => {
     return (
       <Flex style={{ width: '100%', height: '256px' }}>
         <div