Diego Costa avoided bone damage after suffering an ankle injury in Atletico Madrid's 2-1 friendly defeat to Beitar Jerusalem, the club have confirmed.
The club also brought in Alvaro Morata to bolster the forward position, with the striker joining from Chelsea in the winter window.
Atletico faced Israeli side Beitar on Tuesday in a friendly, allowing Costa to make his first appearance since April 6, but he lasted just 28 minutes before suffering what the club describe as a "high-grade sprain" to his left ankle.
"Diego Costa suffered a high-grade sprain in his left ankle during the first part of the game against Beitar Jerusalem," the club announced.
"The striker has undergone radiological tests at a hospital in Jerusalem that have ruled out bone damage.
"After returning to Madrid, more tests will be done to assess the degree of ligament involvement."
| PARTE MÉDICO ACTUALIZADO— Atlético de Madrid (@Atleti) May 21, 2019
A @diegocosta se le han realizado pruebas radiológicas en un hospital de Jerusalén que han descartado una afectación ósea. Tras regresar a Madrid se realizarán más pruebas para valorar el grado de afectación ligamentosa.
Antoine Griezmann scored Atletico's goal in the defeat at Teddi Malcha Stadium in what is expected to be his final appearance for Atleti, the France star having informed the club he wants to leave in the transfer window.
The news of Griezmann's departure comes one year after the Frenchman committed to the club via a documentary, denying Barcelona in the process.
Costa rejoined Atletico Madrid from Chelsea in 2018, having originally played for the club from 2010-14 before moving to Stamford Bridge.
The Spanish striker has scored 11 goals in 29 appearances since rejoining the club, having fired 56 in his original two-season stint with the team.