Adjust test for recent changes in agent model