From 76a881ebc990cbb91a1e74d79f2c2d94161ae831 Mon Sep 17 00:00:00 2001
From: Simon Cruanes <simon.cruanes.2007@m4x.org>
Date: Fri, 30 Aug 2024 13:28:54 -0400
Subject: [PATCH] fix: `LS.get` shouldn't raise

---
 src/core/task_local_storage.ml | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/core/task_local_storage.ml b/src/core/task_local_storage.ml
index 5df4a182..12978926 100644
--- a/src/core/task_local_storage.ml
+++ b/src/core/task_local_storage.ml
@@ -20,8 +20,9 @@ let get_opt k =
     | exception Not_set -> None)
 
 let[@inline] get k ~default =
-  let fiber = get_current_fiber_exn () in
-  PF.FLS.get fiber ~default k
+  match get_current_fiber () with
+  | None -> None
+  | Some fiber -> PF.FLS.get fiber ~default k
 
 let[@inline] set k v : unit =
   let fiber = get_current_fiber_exn () in