diff --git a/lib/src/utils/embeds.dart b/lib/src/utils/embeds.dart
index f7c3c9d0..e25774cb 100644
--- a/lib/src/utils/embeds.dart
+++ b/lib/src/utils/embeds.dart
@@ -2,7 +2,8 @@ import 'dart:math';
 
 import 'package:tuple/tuple.dart';
 
-import '../../flutter_quill.dart';
+import '../models/documents/nodes/leaf.dart';
+import '../widgets/controller.dart';
 
 Tuple2<int, Embed> getEmbedNode(QuillController controller, int offset) {
   var offset = controller.selection.start;