agent: Increase test reliability