diff --git a/tests/unit_tests/test_server_client.py b/tests/unit_tests/test_server_client.py
index 46c64fa7..03de9782 100644
--- a/tests/unit_tests/test_server_client.py
+++ b/tests/unit_tests/test_server_client.py
@@ -2,6 +2,7 @@
 import asyncio
 import datetime
 import json
+import sys
 import uuid
 from asyncio import AbstractEventLoop
 from contextlib import asynccontextmanager, contextmanager
@@ -571,9 +572,6 @@ def test_batch(sync_remote_runnable: RemoteRunnable) -> None:
     )
 
 
-import sys
-
-
 async def test_ainvoke(async_remote_runnable: RemoteRunnable) -> None:
     """Test async invoke."""
     assert await async_remote_runnable.ainvoke(1) == 2