function setAgentStage(address _agent, uint256 _stage) external {
        if (msg.sender == owner() || (msg.sender == agentManager[_agent] && _stage == 1)) {
            Agent(payable(_agent)).setStage(_stage);
        }
    }
