fix: sx props on story

ccontrols · Dec 4, 2020 · ccb1749 · ccb1749
1 parent ade7cd7
commit ccb1749
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/ui/blocks/src/BlockContainer/story/StoryBlockContainer.tsx b/ui/blocks/src/BlockContainer/story/StoryBlockContainer.tsx
@@ -1,4 +1,6 @@
-import React, { FC, useContext } from 'react';
+/** @jsx jsx */
+import { FC, useContext } from 'react';
+import { jsx } from 'theme-ui';
 import { Story } from '@component-controls/core';
 import {
   BlockContainer,
@@ -19,11 +21,11 @@ export const StoryBlockContainer: FC<StoryBlockContainerAllProps> = ({
   story,
   collapsible,
   title: userTitle,
-  sx,
   children,
   useStoryDescription,
   description: userDescription,
   'data-testid': dataTestid,
+  ...rest
 }) => {
   const playground = useContext(PlaygroundContext);
   const title = getStoryBlockTitle({
@@ -39,9 +41,9 @@ export const StoryBlockContainer: FC<StoryBlockContainerAllProps> = ({
       title={title}
       collapsible={collapsible}
       id={userTitle === CURRENT_STORY && story ? story.id : undefined}
-      sx={sx}
       description={description}
       plain={true}
+      {...rest}
     >
       {children}
     </BlockContainer>

diff --git a/ui/blocks/src/Story/Story.tsx b/ui/blocks/src/Story/Story.tsx
@@ -45,7 +45,6 @@ export const Story: FC<StoryProps> = forwardRef(
 
     const { id, name, ...rest } = props;
     const story = useStory({ id, name });
-
     if (story && story.id && story.renderFn) {
       return (
         <StoryBlockContainer {...rest} story={story}>