diff --git a/lib/src/widgets/toolbar/search_dialog.dart b/lib/src/widgets/toolbar/search_dialog.dart
index 84b40ef4..b8ce67ac 100644
--- a/lib/src/widgets/toolbar/search_dialog.dart
+++ b/lib/src/widgets/toolbar/search_dialog.dart
@@ -1,7 +1,9 @@
 import 'package:flutter/material.dart';
 
-import '../../../flutter_quill.dart' hide Text;
 import '../../../translations.dart';
+import '../../models/documents/document.dart';
+import '../../models/themes/quill_dialog_theme.dart';
+import '../controller.dart';
 
 class SearchDialog extends StatefulWidget {
   const SearchDialog(