From 5e73744b12f90d2caa1cda4149dbb6e4e442805a Mon Sep 17 00:00:00 2001
From: Kasper Peulen <kasperpeulen@gmail.com>
Date: Thu, 11 Apr 2024 13:27:56 +0200
Subject: [PATCH] Change type

---
 src/story.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/story.ts b/src/story.ts
index 222b3be..9d60b79 100644
--- a/src/story.ts
+++ b/src/story.ts
@@ -249,11 +249,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> {