diff --git a/src/lib/components/common/RichTextInput.svelte b/src/lib/components/common/RichTextInput.svelte
index be9469cdff..572d5636e0 100644
--- a/src/lib/components/common/RichTextInput.svelte
+++ b/src/lib/components/common/RichTextInput.svelte
@@ -158,16 +158,6 @@
 			this.setupEventListeners();
 		}
 
-		onConnect() {
-			this.isConnected = true;
-			this.joinDocument();
-		}
-
-		onDisconnect() {
-			this.isConnected = false;
-			this.synced = false;
-		}
-
 		setupEventListeners() {
 			// Listen for document updates from server
 			this.socket.on('yjs:document:update', (data) => {
@@ -299,6 +289,16 @@
 			}
 		}
 
+		onConnect() {
+			this.isConnected = true;
+			this.joinDocument();
+		}
+
+		onDisconnect() {
+			this.isConnected = false;
+			this.synced = false;
+		}
+
 		destroy() {
 			this.socket.off('yjs:document:update');
 			this.socket.off('yjs:document:state');
diff --git a/src/lib/components/notes/NoteEditor.svelte b/src/lib/components/notes/NoteEditor.svelte
index 0393a2365d..9f343f5639 100644
--- a/src/lib/components/notes/NoteEditor.svelte
+++ b/src/lib/components/notes/NoteEditor.svelte
@@ -927,7 +927,7 @@ Provide the enhanced notes in markdown format. Use markdown syntax for headings,
 					</div>
 
 					<div
-						class=" flex-1 w-full h-full overflow-auto px-3.5 pb-20 relative z-40 pt-2.5"
+						class=" flex-1 w-full h-full overflow-auto px-3.5 pb-20 relative pt-2.5"
 						id="note-content-container"
 					>
 						{#if enhancing}
@@ -976,14 +976,14 @@ Provide the enhanced notes in markdown format. Use markdown syntax for headings,
 							bind:this={inputElement}
 							bind:editor
 							className="input-prose-sm px-0.5"
+							json={true}
 							bind:value={note.data.content.json}
 							html={note.data?.content?.html}
-							json={true}
-							link={true}
 							documentId={`note:${note.id}`}
 							collaboration={true}
 							socket={$socket}
 							user={$user}
+							link={true}
 							placeholder={$i18n.t('Write something...')}
 							editable={versionIdx === null && !enhancing}
 							onChange={(content) => {
diff --git a/src/lib/components/notes/NoteEditor/Chat.svelte b/src/lib/components/notes/NoteEditor/Chat.svelte
index 1a7e29a7ba..b27e1657b3 100644
--- a/src/lib/components/notes/NoteEditor/Chat.svelte
+++ b/src/lib/components/notes/NoteEditor/Chat.svelte
@@ -281,11 +281,6 @@ Based on the user's instruction, update and enhance the existing notes by incorp
 			scrollToBottom();
 
 			loading = true;
-
-			if (editorEnabled) {
-				insertNoteHandler();
-			}
-
 			await chatCompletionHandler();
 			messages = messages.map((message) => {
 				message.done = true;