diff --git a/pong/static/game.js b/pong/static/game.js
index 5e40a1a..4c985aa 100644
--- a/pong/static/game.js
+++ b/pong/static/game.js
@@ -62,7 +62,7 @@ document.addEventListener('DOMContentLoaded', () => {
                 if (result) {
                     registerForm.style.display = 'none';
                     gameContainer.style.display = 'flex';
-                    startGame();
+                    startWebSocketConnection(token);
                 } else {
                     alert('Registration failed. Please try again.');
                 }