The 22-year-old is making her injury comeback right at the end of the season on Billie Jean King Cup duty