Change type

ComponentDriven · Apr 12, 2024 · 2c2dd28 · 2c2dd28
1 parent 3a2c499
commit 2c2dd28
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/story.ts b/src/story.ts
@@ -260,11 +260,11 @@ export type LoaderFunction<TRenderer extends Renderer = Renderer, TArgs = Args>
 ) => Promise<Record<string, any> | void> | Record<string, any> | void;
 
 type Awaitable<T> = T | PromiseLike<T>;
-export type CleanupCallback = (() => Awaitable<unknown>) | void;
+export type CleanupCallback = () => Awaitable<unknown>;
 
 export type BeforeEach<TRenderer extends Renderer = Renderer, TArgs = Args> = (
   context: StoryContextForLoaders<TRenderer, TArgs>
-) => Awaitable<CleanupCallback>;
+) => Awaitable<CleanupCallback | void>;
 
 export interface StoryContext<TRenderer extends Renderer = Renderer, TArgs = Args>
   extends StoryContextForLoaders<TRenderer, TArgs> {