diff --git a/src/story.ts b/src/story.ts
index e618bfe..222b3be 100644
--- a/src/story.ts
+++ b/src/story.ts
@@ -248,11 +248,12 @@ export type LoaderFunction<TRenderer extends Renderer = Renderer, TArgs = Args>
   context: StoryContextForLoaders<TRenderer, TArgs>
 ) => Promise<Record<string, any> | void> | Record<string, any> | void;
 
-export type Awaitable<T> = T | PromiseLike<T>;
+type Awaitable<T> = T | PromiseLike<T>;
+export type CleanupCallback = (() => Awaitable<unknown>) | void;
 
 export type BeforeEach<TRenderer extends Renderer = Renderer, TArgs = Args> = (
   context: StoryContextForLoaders<TRenderer, TArgs>
-) => Awaitable<(() => Awaitable<unknown>) | void>;
+) => Awaitable<CleanupCallback>;
 
 export interface StoryContext<TRenderer extends Renderer = Renderer, TArgs = Args>
   extends StoryContextForLoaders<TRenderer, TArgs> {