diff --git a/tests/test_type_safety.py b/tests/test_type_safety.py
index 5bfb7ee2..265f273a 100644
--- a/tests/test_type_safety.py
+++ b/tests/test_type_safety.py
@@ -22,7 +22,7 @@ class TestTypeValidation:
             "agent_type": "worker",
             "capabilities": ["compute", "storage"],
             "services": ["task_processing"],
-            "endpoints": ["http://localhost:8001"],
+            "endpoints": {"main": "http://localhost:8001"},
             "metadata": {"version": "1.0.0"}
         }