diff --git a/backend/prompt_studio/prompt_studio_output_manager_v2/output_manager_util.py b/backend/prompt_studio/prompt_studio_output_manager_v2/output_manager_util.py
index 6e945d5e1..b5b6a957d 100644
--- a/backend/prompt_studio/prompt_studio_output_manager_v2/output_manager_util.py
+++ b/backend/prompt_studio/prompt_studio_output_manager_v2/output_manager_util.py
@@ -10,7 +10,7 @@ def get_coverage(
         profile_manager_id: str,
         prompt_id: str = None,
         is_single_pass: bool = False,
-    ) -> dict[str, list[str]]:
+    ) -> list[str]:
         """
         Method to fetch coverage data for given tool and profile manager.