Refactor code

lib/widgets/raw_editor.dart

@@ -919,12 +919,13 @@ class RawEditorState extends EditorState
 
     SchedulerBinding.instance!.addPostFrameCallback(
         (Duration _) => _updateOrDisposeSelectionOverlayIfNeeded());
-    if (!mounted) return;
+    if (mounted) {
       setState(() {
         // Use widget.controller.value in build()
         // Trigger build and updateChildren
       });
     }
+  }
 
   void _updateOrDisposeSelectionOverlayIfNeeded() {
     if (_selectionOverlay != null) {