AgentManagerTest: Fix tests