Update agent_test.go