diff --git a/src/components/problem-editor.tsx b/src/components/problem-editor.tsx
index 1757d6e..bf9dbae 100644
--- a/src/components/problem-editor.tsx
+++ b/src/components/problem-editor.tsx
@@ -60,7 +60,7 @@ export const ProblemEditor = ({
   ) : (
     <CoreEditor
       language={language}
-      value={optimizedCode}
+      value={value}
       path={path}
       languageServerConfigs={languageServerConfigs}
       onEditorReady={setEditor}