diff --git a/packages/graphql_flutter/lib/src/widgets/mutation.dart b/packages/graphql_flutter/lib/src/widgets/mutation.dart
index 140e2bf01..fbf135f25 100644
--- a/packages/graphql_flutter/lib/src/widgets/mutation.dart
+++ b/packages/graphql_flutter/lib/src/widgets/mutation.dart
@@ -10,7 +10,7 @@ typedef RunMutation<TParsed> = MultiSourceResult<TParsed> Function(
 });
 
 typedef MutationBuilder<TParsed> = Widget Function(
-  RunMutation runMutation,
+  RunMutation<TParsed> runMutation,
   QueryResult<TParsed>? result,
 );