diff --git a/jaclang/langserve/tests/test_server.py b/jaclang/langserve/tests/test_server.py
index e656e0ea0..20a0ba49c 100644
--- a/jaclang/langserve/tests/test_server.py
+++ b/jaclang/langserve/tests/test_server.py
@@ -422,8 +422,8 @@ def test_py_type__definition(self) -> None:
         positions = [
             (19, 29, "pygame_mock/color.py:0:0-2:4"),
             (3, 17, "/pygame_mock/__init__.py:0:0-0:0"),
-            (14 + 5, 45, "pygame_mock/color.py:0:0-2:4"),
-            (13 + 6, 77, "mock/constants.py:4:3-4:15"),
+            (20, 45, "pygame_mock/color.py:0:0-2:4"),
+            (19, 77, "mock/constants.py:4:3-4:15"),
             (26, 28, "mock/display.py:0:0-1:7"),
             (24, 22, "/argparse.pyi:124:0-249:13"),
             (19, 74, "pygame_mock/constants.py:4:3-4:15"),
@@ -452,27 +452,41 @@ def test_py_type__references(self) -> None:
         lsp.deep_check(circle_file)
         test_cases = [
             (
-                12,
-                15,
+                2,
+                21,
                 [
                     ":6:21-6:32",
                     ":7:11-7:22",
                     ":11:25-11:36",
                     ":12:15-12:26",
                     ":18:33-18:44",
+                    "19:46-19:57",
                     ":19:8-19:19",
                     ":19:46-19:57",
                     ":20:8-20:19",
+                    "21:8-21:19,",
+                    "23:8-23:19",
                     ":26:4-26:15",
                 ],
             ),
-            (13, 63, ["6:33-6:42", "7:23-7:32", "12:45-12:54", "13:58-13:67"]),
             (
-                15,
+                19,
+                63,
+                [
+                    "6:33-6:42",
+                    "7:23-7:32",
+                    "18:45-18:54",
+                    "19:58-19:67",
+                    "11:37-11:46",
+                    "12:27-12:36",
+                ],
+            ),
+            (
+                24,
                 53,
                 [
-                    "15:42-15:56",
-                    "15:16-15:30",
+                    "24:42-24:56",
+                    "24:16-24:30",
                     "argparse.pyi:334:21-334:35",
                     "argparse.pyi:163:29-163:43",
                     "argparse.pyi:32:52-32:66",