diff --git a/flutter_quill_extensions/lib/embeds/toolbar/formula_button.dart b/flutter_quill_extensions/lib/embeds/toolbar/formula_button.dart
index bf777b67..fb0ab679 100644
--- a/flutter_quill_extensions/lib/embeds/toolbar/formula_button.dart
+++ b/flutter_quill_extensions/lib/embeds/toolbar/formula_button.dart
@@ -1,38 +1,25 @@
 import 'package:flutter/material.dart';
 import 'package:flutter_quill/flutter_quill.dart';
 
-import '../embed_types.dart';
-
 class FormulaButton extends StatelessWidget {
   const FormulaButton({
     required this.icon,
     required this.controller,
     this.iconSize = kDefaultIconSize,
-    this.onImagePickCallback,
     this.fillColor,
-    this.filePickImpl,
-    this.webImagePickImpl,
-    this.mediaPickSettingSelector,
     this.iconTheme,
     this.dialogTheme,
     Key? key,
   }) : super(key: key);
 
   final IconData icon;
+
   final double iconSize;
 
   final Color? fillColor;
 
   final QuillController controller;
 
-  final OnImagePickCallback? onImagePickCallback;
-
-  final WebImagePickImpl? webImagePickImpl;
-
-  final FilePickImpl? filePickImpl;
-
-  final MediaPickSettingSelector? mediaPickSettingSelector;
-
   final QuillIconTheme? iconTheme;
 
   final QuillDialogTheme? dialogTheme;
diff --git a/flutter_quill_extensions/lib/flutter_quill_extensions.dart b/flutter_quill_extensions/lib/flutter_quill_extensions.dart
index eefecbf0..7ef38328 100644
--- a/flutter_quill_extensions/lib/flutter_quill_extensions.dart
+++ b/flutter_quill_extensions/lib/flutter_quill_extensions.dart
@@ -84,10 +84,6 @@ class FlutterQuillEmbeds {
               icon: Icons.functions,
               iconSize: toolbarIconSize,
               controller: controller,
-              onImagePickCallback: onImagePickCallback,
-              filePickImpl: filePickImpl,
-              webImagePickImpl: webImagePickImpl,
-              mediaPickSettingSelector: mediaPickSettingSelector,
               iconTheme: iconTheme,
               dialogTheme: dialogTheme,
             )