diff --git a/agents-api/agents_api/routers/healthz/check_health.py b/agents-api/agents_api/routers/healthz/check_health.py
index 5a466ba39..a031f3a46 100644
--- a/agents-api/agents_api/routers/healthz/check_health.py
+++ b/agents-api/agents_api/routers/healthz/check_health.py
@@ -1,7 +1,7 @@
 import logging
 from uuid import UUID
 
-from ...models.agent.list_agents import list_agents as list_agents_query
+from ...queries.agents.list_agents import list_agents as list_agents_query
 from .router import router