Flatten function since we're using async anyway

src/browser/StorybookPage.ts

@@ -72,6 +72,12 @@ async function fetchStoriesFromWindow(): Promise<StorybookStory[]> {
     );
   }
 
+  await ready(storybookPreview);
+  await storybookPreview.extract();
+
+  const storyStore = storybookPreview.storyStore as StoryStore<Renderer>;
+  return storyStore.raw().map(pickOnlyNecessaryAndSerializableStoryProperties);
+
   // Ensure the storybook preview is "ready", in a way that works in both Storybook 7 and 8.
   // Alternatively, we could drop Storybook 7 support...
   function ready(storybookPreview: PreviewWeb<Renderer>) {
@@ -86,36 +92,26 @@ async function fetchStoriesFromWindow(): Promise<StorybookStory[]> {
     }
   }
 
-  await ready(storybookPreview);
-
-  return storybookPreview.extract().then(() => {
-    // Pick only the properties we need from Storybook's representation of a story.
-    //
-    // This is necessary because Playwright's `page.evaluate` requires return values to be JSON
-    // serializable, so we need to make sure there are no non-serializable things in this object.
-    // There's no telling what Storybook addons people are using, and whether their parameters are
-    // serializable or not.
-    //
-    // See https://github.com/chanzuckerberg/axe-storybook-testing/issues/44 for a bug caused by this.
-    function pickOnlyNecessaryAndSerializableStoryProperties(
-      story: Story,
-    ): StorybookStory {
-      return {
-        id: story.id,
-        name: story.name,
-        title: story.title,
-        parameters: {
-          axe: story.parameters.axe,
-        },
-      };
-    }
-
-    const storyStore = storybookPreview.storyStore as StoryStore<Renderer>;
-
-    return storyStore
-      .raw()
-      .map(pickOnlyNecessaryAndSerializableStoryProperties);
-  });
+  // Pick only the properties we need from Storybook's representation of a story.
+  //
+  // This is necessary because Playwright's `page.evaluate` requires return values to be JSON
+  // serializable, so we need to make sure there are no non-serializable things in this object.
+  // There's no telling what Storybook addons people are using, and whether their parameters are
+  // serializable or not.
+  //
+  // See https://github.com/chanzuckerberg/axe-storybook-testing/issues/44 for a bug caused by this.
+  function pickOnlyNecessaryAndSerializableStoryProperties(
+    story: Story,
+  ): StorybookStory {
+    return {
+      id: story.id,
+      name: story.name,
+      title: story.title,
+      parameters: {
+        axe: story.parameters.axe,
+      },
+    };
+  }
 }
 
 /**