President Trump discussed with Chinese President Xi Jinping the three UCLA men’s basketball players who were arrested last week for shoplifting in Hangzhou, China, and asked for his assistance in resolving their case.

Trump talked with Xi about the arrests during his visit to Beijing last week, according to the Washington Post. UCLA basketball players LiAngelo Ball, Cody Riley and Jalen Hill were arrested Tuesday for allegedly stealing sunglasses from a Louis Vuitton store, according to ESPN. The team was in Hangzhou before leaving for Shanghai on Tuesday, where UCLA played Georgia Tech on Saturday without Ball, Riley or Hill.

White House press secretary Sarah Sanders confirmed to the Post that Trump did discuss the case with his Chinese counterpart during his visit.

Xi told Trump he would look into the case and “ensure that the players be treated fairly and expeditiously,” a U.S. official told the Washington Post.

White House chief of staff John Kelly has contacted the families of the UCLA players and has also spoken with UCLA coach Steve Alford.

According to ESPN, the three UCLA basketball players were released on bail early Wednesday and did not return to the U.S. with the rest of the team.

A source told ESPN the players could remain in Hangzhou for “a week or two,” and said there’s surveillance footage of them stealing from three stores in a shopping center.