Defining selection criteria to improve yield under drought