diff --git a/src/app/endpoints/query.py b/src/app/endpoints/query.py
index fcbb57a11..47e3fc28e 100644
--- a/src/app/endpoints/query.py
+++ b/src/app/endpoints/query.py
@@ -212,6 +212,7 @@ async def query_endpoint_handler(
         moderation_result,
         endpoint_path,
         compaction.original_input if compaction.compacted else None,
+        shield_ids=query_request.shield_ids,
         no_tools=bool(query_request.no_tools),
     )