From 1f42da82fbcd47cef39e6165a24206d6313fe31e Mon Sep 17 00:00:00 2001
From: Jon Salmon <26483285+Jon-Salmon@users.noreply.github.com>
Date: Wed, 28 Sep 2022 00:21:17 +0100
Subject: [PATCH] Revert #962 (#964)

---
 CHANGELOG.md                | 3 +++
 lib/src/widgets/editor.dart | 5 ++---
 pubspec.yaml                | 2 +-
 3 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 416b47ce..d4e8f538 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,3 +1,6 @@
+# [6.0.6+1]
+* Revert 6.0.6
+
 # [6.0.6]
 * Fix wrong custom embed key.
 
diff --git a/lib/src/widgets/editor.dart b/lib/src/widgets/editor.dart
index 38279fa7..5568b8ba 100644
--- a/lib/src/widgets/editor.dart
+++ b/lib/src/widgets/editor.dart
@@ -521,15 +521,14 @@ class QuillEditorState extends State<QuillEditor>
 
     if (builders != null) {
       var _node = node;
-      final type = node.value.type;
 
       // Creates correct node for custom embed
-      if (type == BlockEmbed.customType) {
+      if (node.value.type == BlockEmbed.customType) {
         _node = Embed(CustomBlockEmbed.fromJsonString(node.value.data));
       }
 
       for (final builder in builders) {
-        if (builder.key == type) {
+        if (builder.key == _node.value.type) {
           return builder.build(context, controller, _node, readOnly);
         }
       }
diff --git a/pubspec.yaml b/pubspec.yaml
index cd24007f..5cfb1351 100644
--- a/pubspec.yaml
+++ b/pubspec.yaml
@@ -1,6 +1,6 @@
 name: flutter_quill
 description: A rich text editor supporting mobile and web (Demo App @ bulletjournal.us)
-version: 6.0.6
+version: 6.0.6+1
 #author: bulletjournal
 homepage: https://bulletjournal.us/home/index.html
 repository: https://github.com/singerdmx/flutter-quill