diff --git a/src/hooks/use-problem.ts b/src/hooks/use-problem.ts
index 52d4c39..049e20b 100644
--- a/src/hooks/use-problem.ts
+++ b/src/hooks/use-problem.ts
@@ -39,12 +39,12 @@ export const useProblem = () => {
   const {
     hydrated,
     editor,
-    monacoLanguageClient,
+    webSocket,
     globalLang,
     currentLang,
     currentValue,
     setEditor,
-    setMonacoLanguageClient,
+    setWebSocket,
     setGlobalLang,
     setCurrentLang,
     setCurrentValue,
@@ -111,9 +111,9 @@ export const useProblem = () => {
   return {
     hydrated,
     editor,
-    monacoLanguageClient,
+    webSocket,
     setEditor,
-    setMonacoLanguageClient,
+    setWebSocket,
     globalLang,
     currentLang,
     currentValue,