diff --git a/src/va_agent/agent.py b/src/va_agent/agent.py
index cbccff9..7bd1224 100644
--- a/src/va_agent/agent.py
+++ b/src/va_agent/agent.py
@@ -23,6 +23,7 @@ agent = Agent(
     name=settings.agent_name,
     instruction=settings.agent_instructions,
     tools=[toolset],
+    before_model_callback=governance.before_model_callback,
     after_model_callback=governance.after_model_callback,
 )