diff --git a/code-server-like/static/js/code-like-extension.js b/code-server-like/static/js/code-like-extension.js
index 0b4f32e..6b86c9f 100644
--- a/code-server-like/static/js/code-like-extension.js
+++ b/code-server-like/static/js/code-like-extension.js
@@ -57,12 +57,10 @@ function ChangeWorkspacefileTree(fileTree) {
     }
 }
 
-
-// vscode.window.showInformationMessage('Connecting to server...');
-
-vs_socket = io('vscode');
+const protocol = window.location.protocol === 'https:' ? 'wss:' : 'ws:';
+const host = window.location.host;
+vs_socket = io(`${protocol}//${host}/vscode`);
 vs_socket.on('connect', () => {
-    // vscode.window.showInformationMessage('Connected to server');
     console.log('Connected to server');
     vs_socket.emit('login', { config: code_like_config });
     vs_socket.emit('load_chapter',{config: code_like_config})
diff --git a/code-server-like/static/js/terminal.js b/code-server-like/static/js/terminal.js
index 7166db1..baa03f2 100644
--- a/code-server-like/static/js/terminal.js
+++ b/code-server-like/static/js/terminal.js
@@ -24,7 +24,9 @@
             });
         });
 
-        socket = io('vsc-like',{path:'/vsc-like-ws'});  
+        const protocol = window.location.protocol === 'https:' ? 'wss:' : 'ws:';
+        const host = window.location.host;
+        socket = io(`${protocol}//${host}/vsc-like`, { path: '/vsc-like-ws' });
         const status = document.getElementById("status")
 
         socket.on("pty_output", function(data){