Babar Azam is set to return for the second Test as Lord's as Pakistan seek a response after their innings defeat to England at Headingley.