diff --git a/packages/extension-query/src/MutationRunner.ts b/packages/extension-query/src/MutationRunner.ts
index 0370ab4..0f5d06f 100644
--- a/packages/extension-query/src/MutationRunner.ts
+++ b/packages/extension-query/src/MutationRunner.ts
@@ -19,7 +19,7 @@ export interface MutationRunner<K extends readonly unknown[], A, E, R> {
 
 export interface MakeProps<EH, K extends readonly unknown[], A, E, HandledE, R> {
     readonly QueryClient: QueryClient.GenericTagClass<EH, HandledE>
-    readonly mutation: (key: K) => Effect.Effect<A, E, R>
+    readonly mutation: (key: K) => Effect.Effect<A, E, R | QueryProgress.QueryProgress | QueryState.QueryState<A, Exclude<E, HandledE>>>
 }
 
 export const make = <EH, K extends readonly unknown[], A, E, HandledE, R>(
diff --git a/packages/extension-query/src/QueryExtension.ts b/packages/extension-query/src/QueryExtension.ts
index 1be7476..ef0be44 100644
--- a/packages/extension-query/src/QueryExtension.ts
+++ b/packages/extension-query/src/QueryExtension.ts
@@ -5,8 +5,10 @@ import { ReffuseExtension, type ReffuseHelpers } from "reffuse"
 import * as MutationRunner from "./MutationRunner.js"
 import type * as MutationService from "./MutationService.js"
 import * as QueryClient from "./QueryClient.js"
+import type * as QueryProgress from "./QueryProgress.js"
 import * as QueryRunner from "./QueryRunner.js"
 import type * as QueryService from "./QueryService.js"
+import type * as QueryState from "./QueryState.js"
 
 
 export interface UseQueryProps<K extends readonly unknown[], A, E, R> {
@@ -27,7 +29,7 @@ export interface UseQueryResult<K extends readonly unknown[], A, E> {
 
 
 export interface UseMutationProps<K extends readonly unknown[], A, E, R> {
-    readonly mutation: (key: K) => Effect.Effect<A, E, R>
+    readonly mutation: (key: K) => Effect.Effect<A, E, R | QueryProgress.QueryProgress | QueryState.QueryState<A, E>>
 }
 
 export interface UseMutationResult<K extends readonly unknown[], A, E> {