diff --git a/tests/test_ssh_server.py b/tests/test_ssh_server.py
index 2a94bd0..07a3893 100644
--- a/tests/test_ssh_server.py
+++ b/tests/test_ssh_server.py
@@ -62,10 +62,10 @@ class TestModeSelection:
             mock_menu.return_value = None
             await handle_client(process)
 
-            # Verify mode was 'play'
+            # Verify mode was 'play' (4th arg: process, session, username, mode)
             mock_menu.assert_called_once()
             call_args = mock_menu.call_args
-            assert call_args[0][2] == "play"  # mode argument
+            assert call_args[0][3] == "play"  # mode is 4th positional arg
 
     @pytest.mark.asyncio
     async def test_learn_mode(self):
@@ -85,7 +85,7 @@ class TestModeSelection:
             await handle_client(process)
 
             call_args = mock_menu.call_args
-            assert call_args[0][2] == "tutorial"
+            assert call_args[0][3] == "tutorial"  # mode is 4th positional arg
 
     @pytest.mark.asyncio
     async def test_watch_mode(self):
@@ -105,7 +105,7 @@ class TestModeSelection:
             await handle_client(process)
 
             call_args = mock_menu.call_args
-            assert call_args[0][2] == "spectate"
+            assert call_args[0][3] == "spectate"  # mode is 4th positional arg
 
 
 class TestChessBoard: