update to match the new agent structure