diff --git a/src/lib/components/chat/Messages/CodeBlock.svelte b/src/lib/components/chat/Messages/CodeBlock.svelte
index 0fb6e5c2bf..4ae1a202b9 100644
--- a/src/lib/components/chat/Messages/CodeBlock.svelte
+++ b/src/lib/components/chat/Messages/CodeBlock.svelte
@@ -559,7 +559,7 @@
 						<DiffBlock code={_code} />
 					{:else if edit}
 						<CodeEditor
-							value={isDiff ? _code : code}
+							value={_code}
 							{id}
 							{lang}
 							onSave={() => {