diff --git a/vulture_allowlist.py b/vulture_allowlist.py
index 32d3686b1..a776f47f4 100644
--- a/vulture_allowlist.py
+++ b/vulture_allowlist.py
@@ -232,8 +232,9 @@ NetworkMap
 Arc.dst_machine
 IMitigationsRepository.get_mitigations
 IMitigationsRepository.save_mitigations
-IAgentRepository.save_agent
-IAgentRepository.get_agents
+IAgentRepository.upsert_agent
+IAgentRepository.get_agent_by_id
+IAgentRepository.get_running_agents
 agent
 IAttackRepository.get_attack_report
 IAttackRepository.save_attack_report