Fixed the update_agent_fields function