From 299109d42169318655e48f384d86798c3545e420 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Julien=20Valverd=C3=A9?= <julien.valverde@mailo.com>
Date: Sun, 16 Mar 2025 06:25:02 +0100
Subject: [PATCH] Mutation fix

---
 packages/extension-query/src/MutationRunner.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/packages/extension-query/src/MutationRunner.ts b/packages/extension-query/src/MutationRunner.ts
index 881966c..d8c294e 100644
--- a/packages/extension-query/src/MutationRunner.ts
+++ b/packages/extension-query/src/MutationRunner.ts
@@ -30,7 +30,8 @@ export const make = <EH, K extends readonly unknown[], A, E, HandledE, R>(
 
     const mutate = (...key: K) => QueryClient.pipe(
         Effect.flatMap(client => client.ErrorHandler),
-        Effect.flatMap(errorHandler => mutation(...key).pipe(
+        Effect.flatMap(errorHandler => Ref.set(stateRef, AsyncData.loading()).pipe(
+            Effect.andThen(mutation(...key)),
             errorHandler.handle,
             Effect.tapErrorCause(c => Ref.set(stateRef, AsyncData.failure(c))),
             Effect.tap(v => Ref.set(stateRef, AsyncData.success(v))),