Update test_agent.cpp