fix: sb6 docs compatibility

ccontrols · Oct 14, 2020 · 8d3a767 · 8d3a767
1 parent f37b1ba
commit 8d3a767
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/examples/stories/src/stories/stories-async.stories.tsx b/examples/stories/src/stories/stories-async.stories.tsx
@@ -43,10 +43,9 @@ hooksStory.description =
 hooksStory.decorators = [
   (controls, context) => {
     const { renderFn } = context;
+    const story = typeof renderFn === 'function' ? renderFn : controls;
     return (
-      <div style={{ background: 'lightblue' }}>
-        {renderFn(controls, context)}
-      </div>
+      <div style={{ background: 'lightblue' }}>{story(controls, context)}</div>
     );
   },
 ];
@@ -58,11 +57,11 @@ asyncDecorators.description =
 asyncDecorators.decorators = [
   (controls, context) => {
     const { value } = useAsync(fetchData);
-
     const { renderFn } = context;
+    const story = typeof renderFn === 'function' ? renderFn : controls;
     return (
       <div style={{ background: 'lightpink' }}>
-        {renderFn(controls, { ...context, employee: value })}
+        {story(controls, { ...context, employee: value })}
       </div>
     );
   },
@@ -85,9 +84,10 @@ asyncHooksDecorators.decorators = [
     }, []);
 
     const { renderFn } = context;
+    const story = typeof renderFn === 'function' ? renderFn : controls;
     return (
       <div style={{ background: 'lightpink' }}>
-        {renderFn(controls, { ...context, employee })}
+        {story(controls, { ...context, employee })}
       </div>
     );
   },