Update to use resolveOf

storybookjs · Jan 19, 2023 · 3f1154e · 3f1154e
1 parent 7fa20f2
commit 3f1154e
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/code/ui/blocks/src/blocks/Source.tsx b/code/ui/blocks/src/blocks/Source.tsx
@@ -111,9 +111,7 @@ export const useSourceProps = (
   // The check didn't actually change the type.
   let stories: PreparedStory[] = storiesFromIds as PreparedStory[];
   if (props.of) {
-    const resolved = docsContext.resolveModuleExport(props.of);
-    if (resolved.type !== 'story')
-      throw new Error(`Invalid value passed to the 'of' prop, it should be a story export.`);
+    const resolved = docsContext.resolveOf(props.of, ['story']);
     stories = [resolved.story];
   } else if (stories.length === 0) {
     stories = [docsContext.storyById()];