Messi set to return after two-month lay off