diff --git a/packages/docz-theme-default/src/components/ui/Editor/index.tsx b/packages/docz-theme-default/src/components/ui/Editor/index.tsx
index fd4574b2d..77013b1c9 100644
--- a/packages/docz-theme-default/src/components/ui/Editor/index.tsx
+++ b/packages/docz-theme-default/src/components/ui/Editor/index.tsx
@@ -103,7 +103,7 @@ export class Editor extends Component<EditorProps, EditorState> {
     const editorProps = (config: any) => ({
       value: this.state.code,
       className: editorClassName,
-      editorDidMount: this.removeLastLine,
+      editorDidMount: this.onEditorDidMount,
       onBeforeChange: this.handleChange,
       options: {
         ...options,
@@ -124,6 +124,13 @@ export class Editor extends Component<EditorProps, EditorState> {
     )
   }
 
+  private onEditorDidMount = (editor: any) => {
+    if (editor) {
+      this.removeLastLine(editor)
+      editor.refresh()
+    }
+  }
+
   private removeLastLine = (editor: any) => {
     if (editor && !this.props.withLastLine && this.props.readOnly) {
       const lastLine = editor.lastLine()