Quality Sport Images/Getty Images

Naomi Osaka’s return to the WTA tour is set.

The world No. 2 will play in the National Bank Open in Montreal in August, event organizers announced on Wednesday. Osaka has taken time away from the tour since withdrawing from the French Open and Wimbledon, citing her mental health.

The Montreal event is scheduled to commence on Aug. 9, directly following the Tokyo Olympics, where Osaka will be representing her home country of Japan.

The National Bank Open is a crucial tune-up tournament for the U.S. Open, the season’s last Grand Slam in New York.