Babar Azam set to return for Pakistan in second Test