From 084adcb1e798a7ace469b6192d898c9371680742 Mon Sep 17 00:00:00 2001
From: Xin Yao <singerdmx@gmail.com>
Date: Fri, 2 Apr 2021 21:07:48 -0700
Subject: [PATCH] Refactor code

---
 lib/widgets/raw_editor.dart | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/lib/widgets/raw_editor.dart b/lib/widgets/raw_editor.dart
index 1b649274..6a24f39c 100644
--- a/lib/widgets/raw_editor.dart
+++ b/lib/widgets/raw_editor.dart
@@ -919,11 +919,12 @@ class RawEditorState extends EditorState
 
     SchedulerBinding.instance!.addPostFrameCallback(
         (Duration _) => _updateOrDisposeSelectionOverlayIfNeeded());
-    if (!mounted) return;
-    setState(() {
-      // Use widget.controller.value in build()
-      // Trigger build and updateChildren
-    });
+    if (mounted) {
+      setState(() {
+        // Use widget.controller.value in build()
+        // Trigger build and updateChildren
+      });
+    }
   }
 
   void _updateOrDisposeSelectionOverlayIfNeeded() {