diff --git a/lib/src/form_builder_image_picker.dart b/lib/src/form_builder_image_picker.dart
index 788f8dd..225cd2d 100644
--- a/lib/src/form_builder_image_picker.dart
+++ b/lib/src/form_builder_image_picker.dart
@@ -38,8 +38,12 @@ class FormBuilderImagePicker extends FormBuilderFieldDecoration<List<dynamic>> {
   final EdgeInsetsGeometry? previewMargin;
 
   /// May be supplied for a fully custom display of the image preview
-  final Widget Function(BuildContext, List<Widget> children, Widget? addButton)?
-  previewBuilder;
+  final Widget Function(
+    BuildContext,
+    List<Widget> children,
+    Widget? addButton,
+    FormBuilderImagePickerState state,
+  )? previewBuilder;
 
   /// placeholder image displayed when picking a new image
   final ImageProvider? placeholderImage;
@@ -325,14 +329,14 @@ class FormBuilderImagePicker extends FormBuilderFieldDecoration<List<dynamic>> {
                          .mapIndexed((i, v) => itemBuilder(context, v, i))
                          .toList();
 
-                 return previewBuilder(
-                   context,
-                   widgets,
-                   canUpload ? addButtonBuilder(context) : null,
-                 );
-               },
-             );
-           }
+                return previewBuilder(
+                  context,
+                  widgets,
+                  canUpload ? addButtonBuilder(context) : null,
+                  state,
+                );
+              });
+            }
 
            final child = SizedBox(
              height: previewHeight,