diff --git a/backend/prompt_studio/prompt_studio_core_v2/views.py b/backend/prompt_studio/prompt_studio_core_v2/views.py
index 43d0b0a2a0..f2682ae2f3 100644
--- a/backend/prompt_studio/prompt_studio_core_v2/views.py
+++ b/backend/prompt_studio/prompt_studio_core_v2/views.py
@@ -115,7 +115,7 @@ def get_queryset(self) -> QuerySet | None:
                 ToolStudioPrompt.objects.filter(tool_id=OuterRef("pk"))
                 .order_by()
                 .values("tool_id")
-                .annotate(cnt=Count("id"))
+                .annotate(cnt=Count("prompt_id"))
                 .values("cnt")
             )
             qs = qs.select_related("created_by").annotate(