diff --git a/gateway/integrations/guardrails.py b/gateway/integrations/guardrails.py
index 425f084..bd5846a 100644
--- a/gateway/integrations/guardrails.py
+++ b/gateway/integrations/guardrails.py
@@ -148,6 +148,7 @@ async def check_guardrails(
                     "messages": messages[:-1] + [new_message],
                     "policies": [g.content for g in guardrails],
                     "parameters": context.guardrails_parameters or {},
+                    "dataset_name": context.dataset_name,
                 },
                 headers={
                     "Authorization": context.get_guardrailing_authorization(),