put the with tabs story back the way it was

storybookjs · Apr 8, 2022 · 2317c2a · 2317c2a
1 parent 24c5dc7
commit 2317c2a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 19 deletions.
diff --git a/lib/ui/src/components/preview/preview.stories.tsx b/lib/ui/src/components/preview/preview.stories.tsx
@@ -144,22 +144,7 @@ export const WithCanvasTab = () => (
   </Consumer>
 );
 
-export const WithTabs = () => (
-  <Consumer>
-    {({ api }: Combo) => {
-      return (
-        <Preview
-          {...previewProps}
-          api={{
-            ...api,
-            getElements: () => ({}),
-            getQueryParam: () => '',
-          }}
-        />
-      );
-    }}
-  </Consumer>
-);
+export const WithTabs = () => <Preview {...previewProps} />;
 
 export const WithToolbarExclusions = () => (
   <Consumer>

diff --git a/lib/ui/src/components/preview/toolbar.tsx b/lib/ui/src/components/preview/toolbar.tsx
@@ -132,9 +132,10 @@ const useTools = (
     [defaultToolsExtra, toolsExtraFromConfig]
   );
 
-  const toolbarExclusions = getQueryParam(TOOLBAR_EXCLUSION_PARAM)
-    ? getQueryParam(TOOLBAR_EXCLUSION_PARAM).split(',')
-    : [];
+  const toolbarExclusions: string[] =
+    getQueryParam && getQueryParam(TOOLBAR_EXCLUSION_PARAM)
+      ? getQueryParam(TOOLBAR_EXCLUSION_PARAM).split(',')
+      : [];
 
   return useMemo(() => {
     return story && story.parameters